Clean up

Author: YingboMa

src/structural_transformation/partial_state_selection.jl

@@ -143,11 +143,10 @@ end
 function dummy_derivative_graph!(state::TransformationState, jac=nothing)
     var_eq_matching = complete(pantelides!(state))
     complete!(state.structure)
-    # TODO: remove state when done
-    dummy_derivative_graph!(state.structure, var_eq_matching, jac, state)
+    dummy_derivative_graph!(state.structure, var_eq_matching, jac)
 end
 
-function dummy_derivative_graph!(structure::SystemStructure, var_eq_matching, jac, state)
+function dummy_derivative_graph!(structure::SystemStructure, var_eq_matching, jac)
     @unpack eq_to_diff, var_to_diff, graph = structure
     diff_to_eq = invview(eq_to_diff)
     diff_to_var = invview(var_to_diff)

test/state_selection.jl

@@ -20,7 +20,8 @@ let dd = dummy_derivative(sys)
         has_dx2 |= D(x2) in vars || D(D(x2)) in vars
     end
     @test has_dx1 ⊻ has_dx2 # only one of x1 and x2 can be a dummy derivative
-    @test length(states(dd)) == length(equations(dd)) == 10
+    @test length(states(dd)) == length(equations(dd)) == 9
+    @test length(states(structural_simplify(dd))) <= 6
 end
 
 let pss = partial_state_selection(sys)