test: use new get_u0

Author: AayushSabharwal

test/downstream/test_disturbance_model.jl

@@ -164,7 +164,8 @@ measurement2 = ModelingToolkit.build_explicit_observed_function(
     disturbance_argument = true)
 
 op = ModelingToolkit.inputs(io_sys) .=> 0
-x0, p = ModelingToolkit.get_u0_p(io_sys, op, op)
+x0 = ModelingToolkit.get_u0(io_sys, op)
+p = ModelingToolkit.get_p(io_sys, op)
 x = zeros(5)
 u = zeros(1)
 d = zeros(3)

test/extensions/test_infiniteopt.jl

@@ -62,7 +62,8 @@ InfiniteOpt.@variables(m,
     end)
 
 # Trace the dynamics
-x0, p = ModelingToolkit.get_u0_p(io_sys, [model.θ => 0, model.ω => 0], [model.L => L])
+x0 = ModelingToolkit.get_u0(io_sys, [model.θ => 0, model.ω => 0])
+p = ModelingToolkit.get_p(io_sys, [model.L => L])
 
 xp = f[1](x, u, p, τ)
 cp = f_obs(x, u, p, τ) # Test that it's possible to trace through an observed function

test/initial_values.jl

@@ -8,10 +8,10 @@ using SymbolicIndexingInterface
 @variables x(t)[1:3]=[1.0, 2.0, 3.0] y(t) z(t)[1:2]
 
 @mtkbuild sys=System([D(x) ~ t * x], t) simplify=false
-@test get_u0(sys, [])[1] == [1.0, 2.0, 3.0]
-@test get_u0(sys, [x => [2.0, 3.0, 4.0]])[1] == [2.0, 3.0, 4.0]
-@test get_u0(sys, [x[1] => 2.0, x[2] => 3.0, x[3] => 4.0])[1] == [2.0, 3.0, 4.0]
-@test get_u0(sys, [2.0, 3.0, 4.0])[1] == [2.0, 3.0, 4.0]
+@test get_u0(sys, []) == [1.0, 2.0, 3.0]
+@test get_u0(sys, [x => [2.0, 3.0, 4.0]]) == [2.0, 3.0, 4.0]
+@test get_u0(sys, [x[1] => 2.0, x[2] => 3.0, x[3] => 4.0]) == [2.0, 3.0, 4.0]
+@test get_u0(sys, [2.0, 3.0, 4.0]) == [2.0, 3.0, 4.0]
 
 @mtkbuild sys=System([
         D(x) ~ 3x,
@@ -22,19 +22,19 @@ using SymbolicIndexingInterface
 
 @test_throws ModelingToolkit.MissingVariablesError get_u0(sys, [])
 getter = getu(sys, [x..., y, z...])
-@test getter(get_u0(sys, [y => 4.0, z => [5.0, 6.0]])[1]) == collect(1.0:6.0)
-@test getter(get_u0(sys, [y => 4.0, z => [3y, 4y]])[1]) == [1.0, 2.0, 3.0, 4.0, 12.0, 16.0]
-@test getter(get_u0(sys, [y => 3.0, z[1] => 3y, z[2] => 2x[1]])[1]) ==
+@test getter(get_u0(sys, [y => 4.0, z => [5.0, 6.0]])) == collect(1.0:6.0)
+@test getter(get_u0(sys, [y => 4.0, z => [3y, 4y]])) == [1.0, 2.0, 3.0, 4.0, 12.0, 16.0]
+@test getter(get_u0(sys, [y => 3.0, z[1] => 3y, z[2] => 2x[1]])) ==
       [1.0, 2.0, 3.0, 3.0, 9.0, 2.0]
 
 @variables w(t)
 @parameters p1 p2
 
-@test getter(get_u0(sys, [y => 2p1, z => [3y, 2p2]], [p1 => 5.0, p2 => 6.0])[1]) ==
+@test getter(get_u0(sys, [y => 2p1, z => [3y, 2p2]], [p1 => 5.0, p2 => 6.0])) ==
       [1.0, 2.0, 3.0, 10.0, 30.0, 12.0]
 @test_throws Any getter(get_u0(sys, [y => 2w, w => 3.0, z[1] => 2p1, z[2] => 3p2]))
 @test getter(get_u0(
-    sys, [y => 2w, w => 3.0, z[1] => 2p1, z[2] => 3p2], [p1 => 3.0, p2 => 4.0])[1]) ==
+    sys, [y => 2w, w => 3.0, z[1] => 2p1, z[2] => 3p2], [p1 => 3.0, p2 => 4.0])) ==
       [1.0, 2.0, 3.0, 6.0, 6.0, 12.0]
 
 # Issue#2566