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_empty(self): dag = Dag() self.assertEqual(0, len(dag.nodes())) self.assertEqual(0, len(dag.edges()))
def test_constructor(self): dag = Dag(3, 7, 5) self.assertEqual({3, 5, 7}, dag.nodes())