Merge pull request #1220 from lamorton/vector2

Fixes #1218 issue with get_unit(Vector{Num}).

Author: ChrisRackauckas

src/systems/validation.jl

@@ -20,6 +20,7 @@ end
 equivalent(x,y) = isequal(1*x,1*y)
 unitless = Unitful.unit(1)
 
+get_unit(x::AbstractArray) = map(get_unit,x)
 get_unit(x::Num) = get_unit(value(x))
 function get_unit(x::Symbolic)
     if x isa Sym || operation(x) isa Sym || (operation(x) isa Term && operation(x).f == getindex) || x isa Symbolics.ArrayOp

test/units.jl

@@ -206,4 +206,16 @@ eqs =
 [
     D(x) ~ IfElse.ifelse(t>0.1,2,1)
 ]
-@named sys = ODESystem(eqs, t, vars, [])
+@named sys = ODESystem(eqs, t, vars, [])
+
+#Vectors of symbols
+@parameters t
+@register dummy(vector::Vector{Num}, scalar)
+dummy(vector, scalar) = vector[1] .- scalar
+
+@variables vec[1:2](t)
+vec = collect(vec)
+eqs = [vec .~ dummy(vec, vec[1]);]
+sts = vcat(vec)
+ODESystem(eqs, t, [sts...;], [], name=:sys)
+