Fix tearing reassemble

Author: YingboMa

src/structural_transformation/symbolics_tearing.jl

@@ -602,7 +602,7 @@ function tearing_reassemble(state::TearingState, var_eq_matching, ag = nothing;
     # Contract the vertices in the structure graph to make the structure match
     # the new reality of the system we've just created.
     graph = contract_variables(graph, var_eq_matching, varsperm, eqsperm,
-                               length(solved_variables))
+                               length(solved_variables), length(solved_variables_set))
 
     # Update system
     new_var_to_diff = complete(DiffGraph(length(invvarsperm)))

src/structural_transformation/tearing.jl

@@ -22,7 +22,7 @@ function masked_cumsum!(A::Vector)
 end
 
 function contract_variables(graph::BipartiteGraph, var_eq_matching::Matching,
-                            var_rename, eq_rename, nelim)
+                            var_rename, eq_rename, nelim_eq, nelim_var)
     dig = DiCMOBiGraph{true}(graph, var_eq_matching)
 
     # Update bipartite graph
@@ -31,7 +31,7 @@ function contract_variables(graph::BipartiteGraph, var_eq_matching::Matching,
          for v′ in neighborhood(dig, v, Inf; dir = :in) if var_rename[v′] != 0]
     end
 
-    newgraph = BipartiteGraph(nsrcs(graph) - nelim, ndsts(graph) - nelim)
+    newgraph = BipartiteGraph(nsrcs(graph) - nelim_eq, ndsts(graph) - nelim_var)
     for e in 𝑠vertices(graph)
         ne = eq_rename[e]
         ne == 0 && continue