def test_bft_cyclic(): from simple_graph import SimpleGraph new_graph = SimpleGraph() new_graph.graph = {'A': {'B': 10, 'C': 20}, 'B': {'D': 5, 'E': 3}, 'D': {}, 'E': {'F': 13}, 'C': {'F': 16}, 'F': {'A': 100}, } visited = new_graph.bft('A') keys = list(new_graph.graph.keys()) for val in visited: assert val in keys