fixup! feat: support array variables in linear_expansion

src/linear_algebra.jl

@@ -338,13 +338,34 @@ function _linear_expansion(t, x)
     else
         for (i, arg) in enumerate(args)
             isequal(arg, arrx) && return (0, 0, false)
+            if symbolic_type(arg) == NotSymbolic()
+                arg isa AbstractArray || continue
+                _occursin_array(x, arrx, arg) && return (0, 0, false)
+                continue
+            end
             a, b, islinear = linear_expansion(arg, x)
             (_iszero(a) && islinear) || return (0, 0, false)
         end
         return (0, t, true)
     end
 end
 
+"""
+    _occursin_array(sym, arrsym, arr)
+
+Check if `sym` (or, if `sym` is an element of an array symbolic, the array symbolic
+`arrsym`) occursin in the non-symbolic array `arr`.
+"""
+function _occursin_array(sym, arrsym, arr)
+    for el in arr
+        if symbolic_type(el) == NotSymbolic()
+            return el isa AbstractArray && _occursin_array(sym, arrsym, el)
+        else
+            return occursin(sym, el) || occursin(arrsym, el)
+        end
+    end
+end
+
 ###
 ### Utilities
 ###

test/linear_solver.jl

@@ -70,4 +70,7 @@ a, b, islinear = Symbolics.linear_expansion(D(x) - x, x)
     @test islin && isequal(a, 2) && isequal(b, 0)
     a, b, islin = linear_expansion(y[1], x[1])
     @test islin && isequal(a, 0) && isequal(b, y[1])
+    @test !linear_expansion(z([x...]), x[1])[3]
+    @test !linear_expansion(z(collect(unwrap(x))), x[1])[3]
+    @test !linear_expansion(z([x, 2x]), x[1])[3]
 end