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)
