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)
