more test fixes

Author: vyudu

src/inputoutput.jl

@@ -163,7 +163,7 @@ has_var(ex, x) = x ∈ Set(get_variables(ex))
     (f_oop, f_ip), x_sym, p_sym, io_sys = generate_control_function(
             sys::AbstractODESystem,
             inputs             = unbound_inputs(sys),
-            disturbance_inputs = Any[];
+            disturbance_inputs = disturbances(sys);
             implicit_dae       = false,
             simplify           = false,
         )

test/code_generation.jl

@@ -70,7 +70,7 @@ end
         @parameters p[1:2] (f::Function)(..)
         @named sys = ODESystem(
             [D(x[0]) ~ p[1] * x[0] + x[2], D(x[1]) ~ p[2] * f(x) + x[2]], t)
-        sys, = structural_simplify(sys, inputs = [x[2]], outputs = [])
+        sys = structural_simplify(sys, inputs = [x[2]], outputs = [])
         @test is_parameter(sys, x[2])
         prob = ODEProblem(sys, [x[0] => 1.0, x[1] => 1.0], (0.0, 1.0),
             [p => ones(2), f => sum, x[2] => 2.0])

test/extensions/test_infiniteopt.jl

@@ -24,8 +24,7 @@ end
 model = complete(model)
 inputs = [model.τ]
 outputs = [model.y]
-model, _ = structural_simplify(model, (inputs, outputs))
-
+model = structural_simplify(model; inputs, outputs)
 f, dvs, psym, io_sys = ModelingToolkit.generate_control_function(
     model, split = false)