mirror of
https://github.com/donnemartin/interactive-coding-challenges
synced 2026-01-15 13:58:02 +00:00
@@ -110,9 +110,7 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"metadata": {
|
||||
"collapsed": false
|
||||
},
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"from collections import OrderedDict\n",
|
||||
@@ -165,21 +163,17 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"metadata": {
|
||||
"collapsed": false
|
||||
},
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"# %load test_trie.py\n",
|
||||
"from nose.tools import assert_true\n",
|
||||
"import unittest\n",
|
||||
"\n",
|
||||
"\n",
|
||||
"class TestTrie(object):\n",
|
||||
"class TestTrie(unittest.TestCase): \n",
|
||||
"\n",
|
||||
" def test_trie(self):\n",
|
||||
" print('Test: Remove from empty trie')\n",
|
||||
" trie = Trie()\n",
|
||||
" assert_true(trie.remove('foo') is None)\n",
|
||||
"\n",
|
||||
" print('Test: Insert')\n",
|
||||
" words = ['a', 'at', 'has', 'hat', 'he',\n",
|
||||
@@ -187,11 +181,7 @@
|
||||
" for word in words:\n",
|
||||
" trie.insert(word)\n",
|
||||
" for word in trie.list_words():\n",
|
||||
" assert_true(trie.find(word) is not None)\n",
|
||||
"\n",
|
||||
" # Remove me\n",
|
||||
" # Remove mens\n",
|
||||
" # Remove a\n",
|
||||
" self.assertTrue(trie.find(word) is not None)\n",
|
||||
" \n",
|
||||
" print('Test: Remove me')\n",
|
||||
" trie.remove('me')\n",
|
||||
@@ -199,9 +189,9 @@
|
||||
" words = ['a', 'at', 'has', 'hat', 'he',\n",
|
||||
" 'men', 'mens', 'met']\n",
|
||||
" for word in words:\n",
|
||||
" assert_true(trie.find(word) is not None)\n",
|
||||
" self.assertTrue(trie.find(word) is not None)\n",
|
||||
" for word in words_removed:\n",
|
||||
" assert_true(trie.find(word) is None)\n",
|
||||
" self.assertTrue(trie.find(word) is None)\n",
|
||||
"\n",
|
||||
" print('Test: Remove mens')\n",
|
||||
" trie.remove('mens')\n",
|
||||
@@ -209,9 +199,9 @@
|
||||
" words = ['a', 'at', 'has', 'hat', 'he',\n",
|
||||
" 'men', 'met']\n",
|
||||
" for word in words:\n",
|
||||
" assert_true(trie.find(word) is not None)\n",
|
||||
" self.assertTrue(trie.find(word) is not None)\n",
|
||||
" for word in words_removed:\n",
|
||||
" assert_true(trie.find(word) is None)\n",
|
||||
" self.assertTrue(trie.find(word) is None)\n",
|
||||
"\n",
|
||||
" print('Test: Remove a')\n",
|
||||
" trie.remove('a')\n",
|
||||
@@ -219,9 +209,9 @@
|
||||
" words = ['at', 'has', 'hat', 'he',\n",
|
||||
" 'men', 'met']\n",
|
||||
" for word in words:\n",
|
||||
" assert_true(trie.find(word) is not None)\n",
|
||||
" self.assertTrue(trie.find(word) is not None)\n",
|
||||
" for word in words_removed:\n",
|
||||
" assert_true(trie.find(word) is None)\n",
|
||||
" self.assertTrue(trie.find(word) is None)\n",
|
||||
"\n",
|
||||
" print('Test: Remove has')\n",
|
||||
" trie.remove('has')\n",
|
||||
@@ -229,16 +219,22 @@
|
||||
" words = ['at', 'hat', 'he',\n",
|
||||
" 'men', 'met']\n",
|
||||
" for word in words:\n",
|
||||
" assert_true(trie.find(word) is not None)\n",
|
||||
" self.assertTrue(trie.find(word) is not None)\n",
|
||||
" for word in words_removed:\n",
|
||||
" assert_true(trie.find(word) is None)\n",
|
||||
" self.assertTrue(trie.find(word) is None)\n",
|
||||
"\n",
|
||||
" print('Success: test_trie')\n",
|
||||
"\n",
|
||||
" def test_trie_remove_invalid(self):\n",
|
||||
" print('Test: Remove from empty trie')\n",
|
||||
" trie = Trie()\n",
|
||||
" self.assertTrue(trie.remove('foo') is None) \n",
|
||||
"\n",
|
||||
"\n",
|
||||
"def main():\n",
|
||||
" test = TestTrie()\n",
|
||||
" test.test_trie()\n",
|
||||
" test.assertRaises(KeyError, test.test_trie_remove_invalid)\n",
|
||||
"\n",
|
||||
"\n",
|
||||
"if __name__ == '__main__':\n",
|
||||
@@ -271,9 +267,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
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user