handle symbols and nothing in partition

Author: baggepinnen

src/named_systems2.jl

@@ -957,16 +957,16 @@ function partition(P::NamedStateSpace; u=Symbol[], y=Symbol[],
     # D21 = nothing,
     # D22 = nothing,
 )
-    if isempty(w)
+    if w === nothing || (w isa AbstractArray && isempty(w))
         w = setdiff(P.u, u)
     end
-    if isempty(z)
+    if z === nothing || (z isa AbstractArray && isempty(z))
         z = setdiff(P.y, y)
     end
-    if isempty(u)
+    if u === nothing || (u isa AbstractArray && isempty(u))
         u = setdiff(P.u, w)
     end
-    if isempty(y)
+    if y === nothing || (y isa AbstractArray && isempty(y))
         y = setdiff(P.y, z)
     end
     # u = names2indices(identity.(u), P.u)

test/test_named_systems2.jl

@@ -44,8 +44,8 @@ s3 = named_ss(G3, x = :x, u = :u, y=:y, z=:z, w=:w)
 @test length(s3.y) == 2
 
 s3part = partition(s3, y=:y, z=:z, w=:w, u=:u)
-@test s3part == G3
-@test performance_mapping(s3part) == s3[:z, :w].sys
+@test s3part.sys.sys == G3.sys
+@test performance_mapping(s3part).sys == s3[:z, :w].sys
 
 # Test prefix matching
 G4 = ControlSystemsBase.ssrand(1,2,3)