mirror of
https://github.com/donnemartin/interactive-coding-challenges
synced 2026-01-07 09:58:03 +00:00
@@ -88,21 +88,59 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"metadata": {
|
||||
"collapsed": true
|
||||
},
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"%run ../bst/bst.py\n",
|
||||
"%load ../bst/bst.py"
|
||||
"# %load ../bst/bst.py\n",
|
||||
"class Node(object):\n",
|
||||
"\n",
|
||||
" def __init__(self, data):\n",
|
||||
" self.data = data\n",
|
||||
" self.left = None\n",
|
||||
" self.right = None\n",
|
||||
" self.parent = None\n",
|
||||
"\n",
|
||||
" def __repr__(self):\n",
|
||||
" return str(self.data)\n",
|
||||
"\n",
|
||||
"\n",
|
||||
"class Bst(object):\n",
|
||||
"\n",
|
||||
" def __init__(self, root=None):\n",
|
||||
" self.root = root\n",
|
||||
"\n",
|
||||
" def insert(self, data):\n",
|
||||
" if data is None:\n",
|
||||
" raise TypeError('data cannot be None')\n",
|
||||
" if self.root is None:\n",
|
||||
" self.root = Node(data)\n",
|
||||
" return self.root\n",
|
||||
" else:\n",
|
||||
" return self._insert(self.root, data)\n",
|
||||
"\n",
|
||||
" def _insert(self, node, data):\n",
|
||||
" if node is None:\n",
|
||||
" return Node(data)\n",
|
||||
" if data <= node.data:\n",
|
||||
" if node.left is None:\n",
|
||||
" node.left = self._insert(node.left, data)\n",
|
||||
" node.left.parent = node\n",
|
||||
" return node.left\n",
|
||||
" else:\n",
|
||||
" return self._insert(node.left, data)\n",
|
||||
" else:\n",
|
||||
" if node.right is None:\n",
|
||||
" node.right = self._insert(node.right, data)\n",
|
||||
" node.right.parent = node\n",
|
||||
" return node.right\n",
|
||||
" else:\n",
|
||||
" return self._insert(node.right, data)\n"
|
||||
]
|
||||
},
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"metadata": {
|
||||
"collapsed": false
|
||||
},
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"class BstValidate(Bst):\n",
|
||||
@@ -129,21 +167,18 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"metadata": {
|
||||
"collapsed": false
|
||||
},
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"# %load test_bst_validate.py\n",
|
||||
"from nose.tools import assert_equal\n",
|
||||
"from nose.tools import raises\n",
|
||||
"import unittest\n",
|
||||
"\n",
|
||||
"\n",
|
||||
"class TestBstValidate(object):\n",
|
||||
"class TestBstValidate(unittest.TestCase):\n",
|
||||
"\n",
|
||||
" @raises(Exception)\n",
|
||||
" def test_bst_validate_empty(self):\n",
|
||||
" validate_bst(None)\n",
|
||||
" bst = BstValidate(None)\n",
|
||||
" bst.validate()\n",
|
||||
"\n",
|
||||
" def test_bst_validate(self):\n",
|
||||
" bst = BstValidate(Node(5))\n",
|
||||
@@ -152,7 +187,7 @@
|
||||
" bst.insert(6)\n",
|
||||
" bst.insert(4)\n",
|
||||
" bst.insert(7)\n",
|
||||
" assert_equal(bst.validate(), True)\n",
|
||||
" self.assertEqual(bst.validate(), True)\n",
|
||||
"\n",
|
||||
" bst = BstValidate(Node(5))\n",
|
||||
" left = Node(5)\n",
|
||||
@@ -161,14 +196,14 @@
|
||||
" bst.root.left = left\n",
|
||||
" bst.root.right = right\n",
|
||||
" bst.root.left.right = invalid\n",
|
||||
" assert_equal(bst.validate(), False)\n",
|
||||
" self.assertEqual(bst.validate(), False)\n",
|
||||
"\n",
|
||||
" print('Success: test_bst_validate')\n",
|
||||
"\n",
|
||||
"\n",
|
||||
"def main():\n",
|
||||
" test = TestBstValidate()\n",
|
||||
" test.test_bst_validate_empty()\n",
|
||||
" test.assertRaises(TypeError, test.test_bst_validate_empty)\n",
|
||||
" test.test_bst_validate()\n",
|
||||
"\n",
|
||||
"\n",
|
||||
@@ -202,9 +237,9 @@
|
||||
"name": "python",
|
||||
"nbconvert_exporter": "python",
|
||||
"pygments_lexer": "ipython3",
|
||||
"version": "3.5.0"
|
||||
"version": "3.7.2"
|
||||
}
|
||||
},
|
||||
"nbformat": 4,
|
||||
"nbformat_minor": 0
|
||||
"nbformat_minor": 1
|
||||
}
|
||||
|
||||
@@ -99,9 +99,7 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 1,
|
||||
"metadata": {
|
||||
"collapsed": true
|
||||
},
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"%run ../bst/bst.py"
|
||||
@@ -110,9 +108,7 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 2,
|
||||
"metadata": {
|
||||
"collapsed": false
|
||||
},
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"import sys\n",
|
||||
@@ -147,9 +143,7 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 3,
|
||||
"metadata": {
|
||||
"collapsed": false
|
||||
},
|
||||
"metadata": {},
|
||||
"outputs": [
|
||||
{
|
||||
"name": "stdout",
|
||||
@@ -161,15 +155,14 @@
|
||||
],
|
||||
"source": [
|
||||
"%%writefile test_bst_validate.py\n",
|
||||
"from nose.tools import assert_equal\n",
|
||||
"from nose.tools import raises\n",
|
||||
"import unittest\n",
|
||||
"\n",
|
||||
"\n",
|
||||
"class TestBstValidate(object):\n",
|
||||
"class TestBstValidate(unittest.TestCase):\n",
|
||||
"\n",
|
||||
" @raises(Exception)\n",
|
||||
" def test_bst_validate_empty(self):\n",
|
||||
" validate_bst(None)\n",
|
||||
" bst = BstValidate(None)\n",
|
||||
" bst.validate()\n",
|
||||
"\n",
|
||||
" def test_bst_validate(self):\n",
|
||||
" bst = BstValidate(Node(5))\n",
|
||||
@@ -178,7 +171,7 @@
|
||||
" bst.insert(6)\n",
|
||||
" bst.insert(4)\n",
|
||||
" bst.insert(7)\n",
|
||||
" assert_equal(bst.validate(), True)\n",
|
||||
" self.assertEqual(bst.validate(), True)\n",
|
||||
"\n",
|
||||
" bst = BstValidate(Node(5))\n",
|
||||
" left = Node(5)\n",
|
||||
@@ -187,14 +180,14 @@
|
||||
" bst.root.left = left\n",
|
||||
" bst.root.right = right\n",
|
||||
" bst.root.left.right = invalid\n",
|
||||
" assert_equal(bst.validate(), False)\n",
|
||||
" self.assertEqual(bst.validate(), False)\n",
|
||||
"\n",
|
||||
" print('Success: test_bst_validate')\n",
|
||||
"\n",
|
||||
"\n",
|
||||
"def main():\n",
|
||||
" test = TestBstValidate()\n",
|
||||
" test.test_bst_validate_empty()\n",
|
||||
" test.assertRaises(TypeError, test.test_bst_validate_empty)\n",
|
||||
" test.test_bst_validate()\n",
|
||||
"\n",
|
||||
"\n",
|
||||
@@ -205,9 +198,7 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 4,
|
||||
"metadata": {
|
||||
"collapsed": false
|
||||
},
|
||||
"metadata": {},
|
||||
"outputs": [
|
||||
{
|
||||
"name": "stdout",
|
||||
@@ -238,9 +229,9 @@
|
||||
"name": "python",
|
||||
"nbconvert_exporter": "python",
|
||||
"pygments_lexer": "ipython3",
|
||||
"version": "3.5.0"
|
||||
"version": "3.7.2"
|
||||
}
|
||||
},
|
||||
"nbformat": 4,
|
||||
"nbformat_minor": 0
|
||||
"nbformat_minor": 1
|
||||
}
|
||||
|
||||
@@ -1,12 +1,11 @@
|
||||
from nose.tools import assert_equal
|
||||
from nose.tools import raises
|
||||
import unittest
|
||||
|
||||
|
||||
class TestBstValidate(object):
|
||||
class TestBstValidate(unittest.TestCase):
|
||||
|
||||
@raises(Exception)
|
||||
def test_bst_validate_empty(self):
|
||||
validate_bst(None)
|
||||
bst = BstValidate(None)
|
||||
bst.validate()
|
||||
|
||||
def test_bst_validate(self):
|
||||
bst = BstValidate(Node(5))
|
||||
@@ -15,7 +14,7 @@ class TestBstValidate(object):
|
||||
bst.insert(6)
|
||||
bst.insert(4)
|
||||
bst.insert(7)
|
||||
assert_equal(bst.validate(), True)
|
||||
self.assertEqual(bst.validate(), True)
|
||||
|
||||
bst = BstValidate(Node(5))
|
||||
left = Node(5)
|
||||
@@ -24,16 +23,16 @@ class TestBstValidate(object):
|
||||
bst.root.left = left
|
||||
bst.root.right = right
|
||||
bst.root.left.right = invalid
|
||||
assert_equal(bst.validate(), False)
|
||||
self.assertEqual(bst.validate(), False)
|
||||
|
||||
print('Success: test_bst_validate')
|
||||
|
||||
|
||||
def main():
|
||||
test = TestBstValidate()
|
||||
test.test_bst_validate_empty()
|
||||
test.assertRaises(TypeError, test.test_bst_validate_empty)
|
||||
test.test_bst_validate()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
main()
|
||||
|
||||
Reference in New Issue
Block a user