mirror of
https://github.com/donnemartin/interactive-coding-challenges
synced 2026-01-13 21:08:06 +00:00
Move graph dfs to a class
This commit is contained in:
@@ -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')
|
||||
|
||||
Reference in New Issue
Block a user