def test_remove_node(self): graph = DiGraph() for i in range(10): graph.add_node(i) for i in range(1, 10, 2): graph.remove_node(i) for i in range(1, 10, 2): self.assertEqual(graph.__contains__(i), False)
def test_add_node(self): graph = DiGraph() for i in range(10): graph.add_node(i) graph.add_node(1, (1, 2, 3)) for i in range(10): self.assertEqual(graph.__contains__(i), True) node1 = graph.get_all_v().get(1) self.assertEqual(node1.get_pos(), None)