missed an edge case in assignment checking

mggg · Mar 14, 2019 · 77b6ede · 77b6ede
1 parent 45861bb
commit 77b6ede
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/gerrychain/partition/partition.py b/gerrychain/partition/partition.py
@@ -56,6 +56,9 @@ def _first_time(self, graph, assignment, updaters):
 
     def validate_assignment(self):
         node_names = set(self.graph.nodes)
+        if len(node_names) != sum(len(dist) for dist in self.assignment.parts.values()):
+            return False
+
         assgn_names = set(name for dist in self.assignment.parts.values() for name in dist)
         return node_names == assgn_names