CLEANUP: Special case for unary named tuples in @relation macro.

AlgebraicJulia · Dec 26, 2020 · 759572d · 759572d
1 parent 0179ff4
commit 759572d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/programs/RelationalPrograms.jl b/src/programs/RelationalPrograms.jl
@@ -192,6 +192,7 @@ function parse_relation_call(call)
       (nothing, parse_relation_kw_args(args)...)
     Expr(:tuple) => (nothing, nothing, Symbol[])
     Expr(:tuple, args...) => (nothing, parse_relation_inferred_args(args)...)
+    Expr(:(=), args...) => (nothing, parse_relation_inferred_args([call])...)
 
     _ => error("Invalid syntax in relation $call")
   end

diff --git a/test/programs/RelationalPrograms.jl b/test/programs/RelationalPrograms.jl
@@ -93,6 +93,10 @@ d2 = @relation ((start=u, stop=w) where (u,w,v)) ->
   (E(src=u, tgt=v); E(src=v, tgt=w))
 @test d1 == d2
 
+# Special case: unary named tuple syntax.
+parsed = @relation ((src=v) where (v, w)) -> E(src=v, tgt=w)
+@test parsed[:outer_port_name] == [:src]
+
 # Special case: closed diagram.
 if VERSION >= v"1.5"
   d1 = @relation ((;) where (v,)) -> E(src=v, tgt=v)