def setUp(self):
self.dg = DirectedGraph()
class TestDirectedGraph(unittest.TestCase):
def setUp(self):
self.dg = DirectedGraph()
def test_add_node(self):
self.assertIsNone(self.dg.get_node(42))
self.dg.add_node(42)
self.assertEquals(self.dg.get_node(42).get_id(), 42)
self.dg.add_node(24)
count = 0
for node_id in self.dg.get_node_ids():
self.assertIn(node_id, (42, 24))
count += 1
self.assertEquals(count, 2)
self.assertEquals(self.dg._node_count, 2)
def test_add_edge(self):
self.assertIsNone(self.dg.get_edge(42, 24))
self.dg.add_edge('edge', 42, 24)
self.assertEquals(self.dg.get_edge(42, 24), 'edge')
parent_id, child_id = self.dg.get_edge_node_ids('edge')
self.assertEquals(parent_id, 42)
self.assertEquals(child_id, 24)
