Merge pull request #3015 from aml5600/aml5600/array-units-fix

Dispatch for symbolic array units

Author: ChrisRackauckas

src/systems/unit_check.jl

@@ -69,6 +69,7 @@ get_unit(x::Real) = unitless
 get_unit(x::DQ.AbstractQuantity) = screen_unit(x)
 get_unit(x::AbstractArray) = map(get_unit, x)
 get_unit(x::Num) = get_unit(unwrap(x))
+get_unit(x::Symbolics.Arr) = get_unit(unwrap(x))
 get_unit(op::Differential, args) = get_unit(args[1]) / get_unit(op.x)
 get_unit(op::Difference, args) = get_unit(args[1]) / get_unit(op.t)
 get_unit(op::typeof(getindex), args) = get_unit(args[1])

test/dq_units.jl

@@ -239,3 +239,10 @@ let
     @test mm2units == MT.oneunit(mm2units)
     @test mmunits == mm2units
 end
+
+# test for array variable units https://github.com/SciML/ModelingToolkit.jl/issues/3009
+let
+    @variables x_vec(t)[1:3] [unit = u"1"] x_mat(t)[1:3, 1:3] [unit = u"1"]
+    @test MT.get_unit(x_vec) == u"1"
+    @test MT.get_unit(x_mat) == u"1"
+end