Added binary tree balance check challenge.

This commit is contained in:
Donne Martin
2015-08-06 06:09:30 -04:00
parent bf4222d6e1
commit 53034144d4
5 changed files with 429 additions and 1 deletions

View File

@@ -0,0 +1,30 @@
from nose.tools import assert_equal
class TestCheckBalance(object):
def test_check_balance(self):
node = Node(5)
insert(node, 3)
insert(node, 8)
insert(node, 1)
insert(node, 4)
assert_equal(check_balance(node), True)
node = Node(5)
insert(node, 3)
insert(node, 8)
insert(node, 9)
insert(node, 10)
assert_equal(check_balance(node), False)
print('Success: test_check_balance')
def main():
test = TestCheckBalance()
test.test_check_balance()
if __name__ == '__main__':
main()