Added tree breadth first search challenge.

This commit is contained in:
Donne Martin
2015-08-01 09:26:03 -04:00
parent a6c7f71379
commit 884d8aca80
4 changed files with 416 additions and 0 deletions

View File

@@ -0,0 +1,27 @@
from __future__ import print_function
from nose.tools import assert_equal
class TestBfs(object):
def test_bfs(self):
root = Node(5)
root.insert(2)
root.insert(8)
root.insert(1)
root.insert(3)
with captured_output() as (out, err):
root.bfs(sys.stdout.write)
assert_equal(out.getvalue().strip(), '52813')
print('Success: test_bfs')
def main():
test = TestBfs()
test.test_bfs()
if __name__ == '__main__':
main()