(self)
| 46 | |
| 47 | class TestBFSSearch(unittest.TestCase): |
| 48 | def test_bfs(self): |
| 49 | self.graph = { |
| 50 | 'A': {'B', 'C', 'E'}, |
| 51 | 'B': {'A', 'D', 'F'}, |
| 52 | 'C': {'A', 'G'}, |
| 53 | 'D': {'B'}, |
| 54 | 'F': {'B'}, |
| 55 | 'E': {'A'}, |
| 56 | 'G': {'C'} |
| 57 | } |
| 58 | result = breadth_first_search.search(self.graph, 'A') |
| 59 | self.assertEqual(result, {'A', 'B', 'D', 'F', 'C', 'G', 'E'}) |
| 60 | result = breadth_first_search.search(self.graph, 'G') |
| 61 | self.assertEqual(result, {'G', 'C', 'A', 'B', 'D', 'F', 'E'}) |
| 62 | |
| 63 | |
| 64 | class TestDFSSearch(unittest.TestCase): |