Add block at named

Author: YingboMa

src/systems/abstractsystem.jl

@@ -938,6 +938,36 @@ function _named_idxs(name::Symbol, idxs, call)
     :($name = $map($sym -> $ex, $idxs))
 end
 
+function single_named_expr(expr)
+    name, call = split_assign(expr)
+    if Meta.isexpr(name, :ref)
+        name, idxs = name.args
+        check_name(name)
+        var = gensym(name)
+        ex = quote
+            $var = $(_named(name, call))
+            $name = map(i -> $rename($var, Symbol($(Meta.quot(name)), :_, i)), $idxs)
+        end
+        ex
+    else
+        check_name(name)
+        :($name = $(_named(name, call)))
+    end
+end
+
+function named_expr(expr)
+    if Meta.isexpr(expr, :block)
+        newexpr = Expr(:block)
+        for ex in expr.args
+            ex isa LineNumberNode && continue
+            push!(newexpr.args, single_named_expr(ex))
+        end
+        newexpr
+    else
+        single_named_expr(expr)
+    end
+end
+
 function check_name(name)
     name isa Symbol ||
         throw(Meta.ParseError("The lhs must be a symbol (a) or a ref (a[1:10]). Got $name."))
@@ -946,6 +976,10 @@ end
 """
     @named y = foo(x)
     @named y[1:10] = foo(x)
+    @named begin
+        y[1:10] = foo(x)
+        z = foo(x)
+    end
     @named y 1:10 i -> foo(x*i)  # This is not recommended
 
 Pass the LHS name to the model. When it's calling anything that's not an
@@ -974,20 +1008,7 @@ julia> @named y[1:3] = foo(x)
 ```
 """
 macro named(expr)
-    name, call = split_assign(expr)
-    if Meta.isexpr(name, :ref)
-        name, idxs = name.args
-        check_name(name)
-        var = gensym(name)
-        ex = quote
-            $var = $(_named(name, call))
-            $name = map(i -> $rename($var, Symbol($(Meta.quot(name)), :_, i)), $idxs)
-        end
-        esc(ex)
-    else
-        check_name(name)
-        esc(:($name = $(_named(name, call))))
-    end
+    esc(named_expr(expr))
 end
 
 macro named(name::Symbol, idxs, call)

test/direct.jl

@@ -257,3 +257,9 @@ foo(i; name) = (; i, name)
 @test isequal(goo, [(i = 10, name = Symbol(:goo_, i)) for i in 1:3])
 @named koo 1:3 i->foo(10i)
 @test isequal(koo, [(i = 10i, name = Symbol(:koo_, i)) for i in 1:3])
+@named begin
+    x = foo(12)
+    y[1:3] = foo(13)
+end
+@test isequal(x, (i = 12, name = :x))
+@test isequal(y, [(i = 13, name = Symbol(:y_, i)) for i in 1:3])

test/structural_transformation/bareiss.jl

@@ -16,12 +16,13 @@ function det_bareiss!(M)
 end
 
 @testset "bareiss tests" begin
-# copy gives a dense matrix
-@testset "bareiss tests: $T" for T in (copy, sparse)
-    # matrix determinent pairs
-    for (M, d) in ((BigInt[9 1 8 0; 0 0 8 7; 7 6 8 3; 2 9 7 7], -1),
-                   (BigInt[1 big(2)^65+1; 3 4], 4 - 3 * (big(2)^65 + 1)))
-        # test that the determinent was correctly computed
-        @test det_bareiss!(T(M)) == d
+    # copy gives a dense matrix
+    @testset "bareiss tests: $T" for T in (copy, sparse)
+        # matrix determinent pairs
+        for (M, d) in ((BigInt[9 1 8 0; 0 0 8 7; 7 6 8 3; 2 9 7 7], -1),
+                       (BigInt[1 big(2)^65+1; 3 4], 4 - 3 * (big(2)^65 + 1)))
+            # test that the determinent was correctly computed
+            @test det_bareiss!(T(M)) == d
+        end
     end
-end end
+end