def test_bfs_visit(self): def visit(node): visited.add(node) visited = set() g = Graph([v, w, x, y], [vw, vx, wx]) g.bfs(v, visit) assert set([v, w, x]) == visited
def test_bfs(self): g = Graph([v, w, x, y], [vw, vx, wx]) assert set([v, w, x]) == g.bfs(v) assert set([v, w, x]) == g.bfs(w) assert set([y, ]) == g.bfs(y)