class GraphTraversalTest(TestCase): """Graph traversal tests""" def setUp(self): self.tree = Graph(v=['A', 'B', 'C', 'D', 'E', 'F', 'G'], e=[('A', ['B', 'C']), ('B', ['D', 'F']), ('E', ['F', 'G'])]) def test_graph_simple_traversal(self): self.assertEqual(['A', 'B', 'C', 'D', 'E', 'F', 'G'], list(self.tree.traversal())) def test_graph_bfs_traversal(self): self.assertEqual(['A', 'B', 'C', 'D', 'E', 'F', 'G'], list(self.tree.traversal(strategy='bfs'))) def test_graph_dfs_traversal(self): self.assertEqual(['G', 'F', 'E', 'D', 'C', 'B', 'A'], list(self.tree.traversal(strategy='dfs')))