update remove_nodes tests for better coverage

DLWoodruff · Apr 3, 2024 · c7db40f · c7db40f
1 parent 025e429
commit c7db40f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pyomo/contrib/incidence_analysis/tests/test_interface.py b/pyomo/contrib/incidence_analysis/tests/test_interface.py
@@ -1466,7 +1466,10 @@ def test_remove_bad_node(self):
         with self.assertRaisesRegex(KeyError, "does not exist"):
             # Suppose we think something like this should work. We should get
             # an error, and not silently do nothing.
-            igraph.remove_nodes([m.x], [m.eq])
+            igraph.remove_nodes([m.x], [m.eq[1]])
+
+        with self.assertRaisesRegex(KeyError, "does not exist"):
+            igraph.remove_nodes(None, [m.eq])
 
         with self.assertRaisesRegex(KeyError, "does not exist"):
             igraph.remove_nodes([[m.x[1], m.x[2]], [m.eq[1]]])