Fix consistency checks

Author: YingboMa

src/bipartite_graph.jl

@@ -68,10 +68,18 @@ end
 
 function complete(m::Matching{U}) where {U}
     m.inv_match !== nothing && return m
-    inv_match = Union{U, Int}[unassigned for _ = 1:length(m.match)]
+    N = length(m.match)
+    inv_match = Union{U, Int}[unassigned for _ = 1:N]
     for (i, eq) in enumerate(m.match)
         isa(eq, Int) || continue
-        inv_match[eq] = i
+        if eq <= N
+            inv_match[eq] = i
+        else # this only happens when the system is imbalanced
+            for k in N+1:eq-1
+                push!(inv_match, unassigned)
+            end
+            push!(inv_match, i)
+        end
     end
     return Matching{U}(collect(m.match), inv_match)
 end

src/structural_transformation/utils.jl

@@ -59,7 +59,7 @@ function check_consistency(state::TearingState)
         iseqs = n_highest_vars < neqs
         if iseqs
             eq_var_matching = invview(complete(var_eq_matching)) # extra equations
-            bad_idxs = findall(isnothing, @view eq_var_matching[1:nsrcs(graph)])
+            bad_idxs = findall(isequal(unassigned), @view eq_var_matching[1:nsrcs(graph)])
         else
             bad_idxs = findall(isequal(unassigned), var_eq_matching)
         end