diff --git a/tests/digraph/test_nodes.py b/tests/digraph/test_nodes.py index d8555feb02..c61640c566 100644 --- 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"])