jlengrand/Coding4Interviews
1
0
Fork 0
mirror of https://github.com/jlengrand/Coding4Interviews.git synced 2026-03-10 08:11:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

enhances print node to take care of more special cases

Browse Source
This commit is contained in:
Julien Lengrand-Lambert
2013-07-12 21:21:20 +02:00
parent cf753f8ac4
commit e731857f82
2 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
01_binary_search_trees/bst.py
View File

@@ -134,4 +134,13 @@ class BinarySearchNode():
"""
prints a single tree node.
"""
return (str(self.left_child.value) + " " + str(self.value) + " " + str(self.right_child.value) )
ret = ""
if self.has_left_child():
ret += str(self.left_child.value) + " "
ret += str(self.value)
if self.has_right_child():
ret += " " + str(self.right_child.value)
return ret

4
01_binary_search_trees/bst_test.py
View File

@@ -208,5 +208,9 @@ class test_binary_search_tree(unittest.TestCase):
self.assertEqual(bst.__str__(), "4 6 10")
self.assertEqual(bst.root_node.right_child.__str__(), "10 16")
self.assertEqual(bst.root_node.right_child.right_child.__str__(), "16")
if __name__ == '__main__':
unittest.main()