def testDepthFirstSearch(self): graph = DictGraph() graph.addEdge(0, 1) graph.addEdge(1, 2) graph.addEdge(1, 3) graph.addEdge(2, 6) graph.addEdge(4, 5) self.assertEquals(graph.depthFirstSearch(0), [0,1,2,6,3]) self.assertEquals(graph.depthFirstSearch(1), [1,0,2,6,3]) self.assertEquals(graph.depthFirstSearch(6), [6,2,1,0,3]) self.assertEquals(graph.depthFirstSearch(4), [4, 5]) self.assertEquals(graph.depthFirstSearch(5), [5, 4])