GitHub/interactive-coding-challenges
1
0
Fork 0
mirror of https://github.com/donnemartin/interactive-coding-challenges synced 2026-03-04 14:48:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move bst to a class

Browse Source
This commit is contained in:
Donne Martin
2016-08-14 08:20:06 -04:00
parent be5f29f98e
commit 82c9725000
4 changed files with 119 additions and 81 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
graphs_trees/bst/bst_challenge.ipynb
View File

@@ -98,9 +98,11 @@
" pass\n",
"\n",
"\n",
"def insert(root, data):\n",
" # TODO: Implement me\n",
" pass"
"class Bst(object):\n",
"\n",
" def insert(self, data):\n",
" # TODO: Implement me\n",
" pass"
]
},
{
@@ -156,22 +158,25 @@
" def __init__(self):\n",
" self.results = Results()\n",
"\n",
" def test_tree(self):\n",
" node = Node(5)\n",
" assert_equal(insert(node, 2).data, 2)\n",
" assert_equal(insert(node, 8).data, 8)\n",
" assert_equal(insert(node, 1).data, 1)\n",
" assert_equal(insert(node, 3).data, 3)\n",
" in_order_traversal(node, self.results.add_result)\n",
" def test_tree_one(self):\n",
" bst = Bst()\n",
" bst.insert(5)\n",
" bst.insert(2)\n",
" bst.insert(8)\n",
" bst.insert(1)\n",
" bst.insert(3)\n",
" in_order_traversal(bst.root, self.results.add_result)\n",
" assert_equal(str(self.results), '[1, 2, 3, 5, 8]')\n",
" self.results.clear_results()\n",
"\n",
" node = insert(None, 1)\n",
" assert_equal(insert(node, 2).data, 2)\n",
" assert_equal(insert(node, 3).data, 3)\n",
" assert_equal(insert(node, 4).data, 4)\n",
" insert(node, 5)\n",
" in_order_traversal(node, self.results.add_result)\n",
" def test_tree_two(self):\n",
" bst = Bst()\n",
" bst.insert(1)\n",
" bst.insert(2)\n",
" bst.insert(3)\n",
" bst.insert(4)\n",
" bst.insert(5)\n",
" in_order_traversal(bst.root, self.results.add_result)\n",
" assert_equal(str(self.results), '[1, 2, 3, 4, 5]')\n",
"\n",
" print('Success: test_tree')\n",
@@ -179,7 +184,8 @@
"\n",
"def main():\n",
" test = TestTree()\n",
" test.test_tree()\n",
" test.test_tree_one()\n",
" test.test_tree_two()\n",
"\n",
"\n",
"if __name__ == '__main__':\n",