Add tests

Qiskit · Apr 22, 2024 · b85915c · b85915c
1 parent 2c8af92
commit b85915c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/digraph/test_nodes.py b/tests/digraph/test_nodes.py
@@ -85,6 +85,7 @@ def test_remove_nodes_retain_edges_single_edge(self):
         res = dag.nodes()
         self.assertEqual(["a", "c"], res)
         self.assertEqual([0, 2], dag.node_indexes())
+        self.assertTrue(dag.has_node(node_a))
         self.assertTrue(dag.has_edge(node_a, node_c))
         self.assertEqual(dag.get_all_edge_data(node_a, node_c), ["Edgy"])
 
@@ -97,6 +98,7 @@ def test_remove_nodes_retain_edges_single_edge_outgoing_weight(self):
         res = dag.nodes()
         self.assertEqual(["a", "c"], res)
         self.assertEqual([0, 2], dag.node_indexes())
+        self.assertTrue(dag.has_node(node_a))
         self.assertTrue(dag.has_edge(node_a, node_c))
         self.assertEqual(dag.get_all_edge_data(node_a, node_c), ["Edgy_mk2"])