def test_two_nodes(self):
dag = Dag()
n0 = dag.add_node(3)
n1 = dag.add_node(5)
self.assertEqual({3, 5}, dag.nodes())
self.assertIs(n0, dag[3])
self.assertIs(n1, dag[5])
def test_one_node(self):
dag = Dag()
node = dag.add_node(3)
self.assertEqual({3}, dag.nodes())
self.assertEqual(set(), dag.edges())
self.assertEqual(3, node.context)
self.assertEqual(dag, node.graph)
self.assertIs(node, dag[3])
def test_add_same_node(self):
dag = Dag()
n = dag.add_node(3)
self.assertIs(n, dag.add_node(3))