Parse an expression of a tuple to Variables

Author: YingboMa

src/variables.jl

@@ -49,11 +49,12 @@ function _parse_vars(macroname, known, x)
     #     y
     #     z
     # end
+    x = x isa Tuple && first(x) isa Expr && first(x).head == :tuple ? first(x).args : x # tuple handling
     x = flatten_expr!(x)
     for _var in x
         iscall = isa(_var, Expr) && _var.head == :call
         issym    = _var isa Symbol
-        @assert iscall || issym "@$macroname expects a tuple of expressions (`@$macroname x y z(t)`)"
+        @assert iscall || issym "@$macroname expects a tuple of expressions or an expression of a tuple (`@$macroname x y z(t)` or `@$macroname x, y, z(t)`)"
 
         if iscall
             dependents = :(Variable[$(_var.args[2:end]...)])
@@ -71,8 +72,8 @@ function _parse_vars(macroname, known, x)
     return ex
 end
 macro variables(xs...)
-    esc(_parse_vars(:Variable, false, xs))
+    esc(_parse_vars(:variables, false, xs))
 end
 macro parameters(xs...)
-    esc(_parse_vars(:Param, true, xs))
+    esc(_parse_vars(:parameters, true, xs))
 end

test/variable_parsing.jl

@@ -2,9 +2,8 @@ using ModelingToolkit
 using Test
 
 @parameters t
-@variables x(t)
-@variables y(t)
-@variables z(t)
+@variables x(t) y(t) # test multi-arg
+@variables z(t) # test single-arg
 x1 = Variable(:x, [t])
 y1 = Variable(:y, [t])
 z1 = Variable(:z, [t])
@@ -33,3 +32,6 @@ D1 = Differential(t)
 @test convert(Expr, D) == D
 
 @test isequal(x ≤ y + 1, (x < y + 1) | (x == y + 1))
+
+@test @macroexpand(@parameters x, y, z(t)) == @macroexpand(@parameters x y z(t))
+@test @macroexpand(@variables x, y, z(t)) == @macroexpand(@variables x y z(t))