def test_remove_node(self):
g = DiGraph()
g.add_node(0)
self.assertEqual(repr(g.get_nodes()[0]), repr(Node(0, None)))
self.assertEqual(len(g.get_nodes()), 1)
g.remove_node(0)
self.assertEqual(len(g.get_nodes()), 0)
def test_add_node(self):
g = DiGraph()
g.add_node(0)
self.assertEqual(repr(g.get_nodes()[0]), repr(Node(0, None)))
g.add_node(1, (1, 2, 3))
self.assertEqual(repr(g.get_nodes()[1]), repr(Node(1, (1, 2, 3))))
g.add_node(2)
self.assertEqual(len(g.get_nodes()), 3)