Simpilfied bst challenge.

This commit is contained in:
Donne Martin
2015-08-01 17:44:14 -04:00
parent e55c7ee1c6
commit 31058abf3a
5 changed files with 106 additions and 63 deletions

View File

@@ -5,24 +5,24 @@ from nose.tools import assert_equal
class TestTree(object):
def test_tree(self):
root = Node(5)
root.insert(2)
root.insert(8)
root.insert(1)
root.insert(3)
node = Node(5)
insert(node, 2)
insert(node, 8)
insert(node, 1)
insert(node, 3)
with captured_output() as (out, err):
root.in_order_traversal(sys.stdout.write)
in_order_traversal(node, sys.stdout.write)
assert_equal(out.getvalue().strip(), '12358')
root = Node(1)
root.insert(2)
root.insert(3)
root.insert(4)
root.insert(5)
node = Node(1)
insert(node, 2)
insert(node, 3)
insert(node, 4)
insert(node, 5)
with captured_output() as (out, err):
root.in_order_traversal(sys.stdout.write)
in_order_traversal(node, sys.stdout.write)
assert_equal(out.getvalue().strip(), '12345')
print('Success: test_tree')