Move bst validate to a class (#98)

This commit is contained in:
Donne Martin
2016-08-30 06:06:58 -04:00
committed by GitHub
parent 982580d051
commit dfeb7f4d80
3 changed files with 59 additions and 55 deletions

View File

@@ -103,9 +103,11 @@
},
"outputs": [],
"source": [
"def validate_bst(node):\n",
" # TODO: Implement me\n",
" pass"
"class BstValidate(Bst):\n",
"\n",
" def validate(self):\n",
" # TODO: Implement me\n",
" pass"
]
},
{
@@ -142,22 +144,22 @@
" validate_bst(None)\n",
"\n",
" def test_bst_validate(self):\n",
" node = Node(5)\n",
" insert(node, 8)\n",
" insert(node, 5)\n",
" insert(node, 6)\n",
" insert(node, 4)\n",
" insert(node, 7)\n",
" assert_equal(validate_bst(node), True)\n",
" bst = BstValidate(Node(5))\n",
" bst.insert(8)\n",
" bst.insert(5)\n",
" bst.insert(6)\n",
" bst.insert(4)\n",
" bst.insert(7)\n",
" assert_equal(bst.validate(), True)\n",
"\n",
" root = Node(5)\n",
" bst = BstValidate(Node(5))\n",
" left = Node(5)\n",
" right = Node(8)\n",
" invalid = Node(20)\n",
" root.left = left\n",
" root.right = right\n",
" root.left.right = invalid\n",
" assert_equal(validate_bst(root), False)\n",
" bst.root.left = left\n",
" bst.root.right = right\n",
" bst.root.left.right = invalid\n",
" assert_equal(bst.validate(), False)\n",
"\n",
" print('Success: test_bst_validate')\n",
"\n",