| 63 | |
| 64 | class TestDFSSearch(unittest.TestCase): |
| 65 | def test_dfs(self): |
| 66 | self.graph = { |
| 67 | 'A': ['B', 'C', 'E'], |
| 68 | 'B': ['A', 'D', 'F'], |
| 69 | 'C': ['A', 'G'], |
| 70 | 'D': ['B'], |
| 71 | 'F': ['B'], |
| 72 | 'E': ['A'], |
| 73 | 'G': ['C'] |
| 74 | } |
| 75 | result = depth_first_search.search(self.graph, 'A') |
| 76 | self.assertEqual(result, ['A', 'B', 'D', 'F', 'C', 'G', 'E']) |
| 77 | result = depth_first_search.search(self.graph, 'G') |
| 78 | self.assertEqual(result, ['G', 'C', 'A', 'B', 'D', 'F', 'E']) |
| 79 | |
| 80 | class TestExponentialSearch(TestSearchingAlgorithm): |
| 81 | def test_exponential_search(self): |