GitHub/interactive-coding-challenges
1
0
Fork 0
mirror of https://github.com/donnemartin/interactive-coding-challenges synced 2026-01-15 13:58:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move tree check balance to a class (#99)

Browse Source
This commit is contained in:
Donne Martin
2016-09-07 05:46:47 -04:00
committed by GitHub
parent dfeb7f4d80
commit d1d1abd3db
3 changed files with 98 additions and 89 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
graphs_trees/check_balance/check_balance_challenge.ipynb
View File

@@ -92,9 +92,11 @@
},
"outputs": [],
"source": [
"def check_balance(root):\n",
" # TODO: Implement me\n",
" pass"
"class BstBalance(Bst):\n",
"\n",
" def check_balance(self):\n",
" # TODO: Implement me\n",
" pass"
]
},
{
@@ -128,33 +130,34 @@
"\n",
" @raises(Exception)\n",
" def test_check_balance_empty(self):\n",
" check_balance(None)\n",
" bst = BstBalance(None)\n",
" bst.check_balance()\n",
"\n",
" def test_check_balance(self):\n",
" node = Node(5)\n",
" assert_equal(check_balance(node), True)\n",
" bst = BstBalance(Node(5))\n",
" assert_equal(bst.check_balance(), True)\n",
"\n",
" insert(node, 3)\n",
" insert(node, 8)\n",
" insert(node, 1)\n",
" insert(node, 4)\n",
" assert_equal(check_balance(node), True)\n",
" bst.insert(3)\n",
" bst.insert(8)\n",
" bst.insert(1)\n",
" bst.insert(4)\n",
" assert_equal(bst.check_balance(), True)\n",
"\n",
" node = Node(5)\n",
" insert(node, 3)\n",
" insert(node, 8)\n",
" insert(node, 9)\n",
" insert(node, 10)\n",
" assert_equal(check_balance(node), False)\n",
" bst = BstBalance(Node(5))\n",
" bst.insert(3)\n",
" bst.insert(8)\n",
" bst.insert(9)\n",
" bst.insert(10)\n",
" assert_equal(bst.check_balance(), False)\n",
"\n",
" node = Node(3)\n",
" insert(node, 2)\n",
" insert(node, 1)\n",
" insert(node, 5)\n",
" insert(node, 4)\n",
" insert(node, 6)\n",
" insert(node, 7)\n",
" assert_equal(check_balance(node), False)\n",
" bst = BstBalance(Node(3))\n",
" bst.insert(2)\n",
" bst.insert(1)\n",
" bst.insert(5)\n",
" bst.insert(4)\n",
" bst.insert(6)\n",
" bst.insert(7)\n",
" assert_equal(bst.check_balance(), True)\n",
"\n",
" print('Success: test_check_balance')\n",
"\n",