def tc1(): """ 0 -> 5,6 5 -> 7 7 -> 10 BFS from 0 to 10 should have a distance of 3 """ nodes = [node(i) for i in xrange(11)] nodes[0].addNeighbors(filter(lambda x: x.name == 5 or x.name == 6, nodes)) nodes[5].addNeighbors(filter(lambda x: x.name == 7, nodes)) nodes[7].addNeighbors(filter(lambda x: x.name == 10, nodes)) dist = bfs(nodes[0], nodes[10]) assert dist == 3
def tc1(): """ 0 -> 5,6 5 -> 7 7 -> 10 BFS from 0 to 10 should have a distance of 3 """ nodes = [node(i) for i in xrange(11)] nodes[0].addNeighbors(filter(lambda x: x.name== 5 or x.name == 6, nodes)) nodes[5].addNeighbors(filter(lambda x:x.name == 7, nodes)) nodes[7].addNeighbors(filter(lambda x:x.name == 10, nodes)) dist = bfs(nodes[0], nodes[10]) assert dist == 3
def tc1(): """ 0 -> 5,6 5 -> 7 7 -> 10 BFS from 0 to 10 should have a distance of 3 """ nodes = [node(i) for i in xrange(11)] weights = {(nodes[i],nodes[j]): 1 for i in xrange(11) for j in xrange(11)} nodes[0].addNeighbors(filter(lambda x: x.name== 5 or x.name == 6, nodes)) nodes[5].addNeighbors(filter(lambda x:x.name == 7, nodes)) nodes[7].addNeighbors(filter(lambda x:x.name == 10, nodes)) dist = dijkstra(weights, nodes[0], nodes[10]) assert dist == 3