Move graph dfs to a class

This commit is contained in:
Donne Martin
2016-09-10 07:41:23 -04:00
parent e1bf3e791a
commit 62f806100e
3 changed files with 21 additions and 17 deletions

View File

@@ -8,7 +8,7 @@ class TestDfs(object):
def test_dfs(self):
nodes = []
graph = Graph()
graph = GraphDfs()
for id in range(0, 6):
nodes.append(graph.add_node(id))
graph.add_edge(0, 1, 5)
@@ -19,7 +19,7 @@ class TestDfs(object):
graph.add_edge(2, 1, 6)
graph.add_edge(3, 2, 7)
graph.add_edge(3, 4, 8)
dfs(nodes[0], self.results.add_result)
graph.dfs(nodes[0], self.results.add_result)
assert_equal(str(self.results), "[0, 1, 3, 2, 4, 5]")
print('Success: test_dfs')