mirror of
https://github.com/donnemartin/interactive-coding-challenges
synced 2026-01-05 00:48:03 +00:00
@@ -1,8 +1,7 @@
|
||||
from nose.tools import assert_true
|
||||
from nose.tools import raises
|
||||
import unittest
|
||||
|
||||
|
||||
class TestTrie(object):
|
||||
class TestTrie(unittest.TestCase):
|
||||
|
||||
def test_trie(self):
|
||||
trie = Trie()
|
||||
@@ -13,7 +12,7 @@ class TestTrie(object):
|
||||
for word in words:
|
||||
trie.insert(word)
|
||||
for word in trie.list_words():
|
||||
assert_true(trie.find(word) is not None)
|
||||
self.assertTrue(trie.find(word) is not None)
|
||||
|
||||
print('Test: Remove me')
|
||||
trie.remove('me')
|
||||
@@ -21,9 +20,9 @@ class TestTrie(object):
|
||||
words = ['a', 'at', 'has', 'hat', 'he',
|
||||
'men', 'mens', 'met']
|
||||
for word in words:
|
||||
assert_true(trie.find(word) is not None)
|
||||
self.assertTrue(trie.find(word) is not None)
|
||||
for word in words_removed:
|
||||
assert_true(trie.find(word) is None)
|
||||
self.assertTrue(trie.find(word) is None)
|
||||
|
||||
print('Test: Remove mens')
|
||||
trie.remove('mens')
|
||||
@@ -31,9 +30,9 @@ class TestTrie(object):
|
||||
words = ['a', 'at', 'has', 'hat', 'he',
|
||||
'men', 'met']
|
||||
for word in words:
|
||||
assert_true(trie.find(word) is not None)
|
||||
self.assertTrue(trie.find(word) is not None)
|
||||
for word in words_removed:
|
||||
assert_true(trie.find(word) is None)
|
||||
self.assertTrue(trie.find(word) is None)
|
||||
|
||||
print('Test: Remove a')
|
||||
trie.remove('a')
|
||||
@@ -41,9 +40,9 @@ class TestTrie(object):
|
||||
words = ['at', 'has', 'hat', 'he',
|
||||
'men', 'met']
|
||||
for word in words:
|
||||
assert_true(trie.find(word) is not None)
|
||||
self.assertTrue(trie.find(word) is not None)
|
||||
for word in words_removed:
|
||||
assert_true(trie.find(word) is None)
|
||||
self.assertTrue(trie.find(word) is None)
|
||||
|
||||
print('Test: Remove has')
|
||||
trie.remove('has')
|
||||
@@ -51,24 +50,23 @@ class TestTrie(object):
|
||||
words = ['at', 'hat', 'he',
|
||||
'men', 'met']
|
||||
for word in words:
|
||||
assert_true(trie.find(word) is not None)
|
||||
self.assertTrue(trie.find(word) is not None)
|
||||
for word in words_removed:
|
||||
assert_true(trie.find(word) is None)
|
||||
self.assertTrue(trie.find(word) is None)
|
||||
|
||||
print('Success: test_trie')
|
||||
|
||||
@raises(Exception)
|
||||
def test_trie_remove_invalid(self):
|
||||
print('Test: Remove from empty trie')
|
||||
trie = Trie()
|
||||
assert_true(trie.remove('foo') is None)
|
||||
self.assertTrue(trie.remove('foo') is None)
|
||||
|
||||
|
||||
def main():
|
||||
test = TestTrie()
|
||||
test.test_trie()
|
||||
test.test_trie_remove_invalid()
|
||||
test.assertRaises(KeyError, test.test_trie_remove_invalid)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
main()
|
||||
|
||||
@@ -72,5 +72,5 @@ class Trie(object):
|
||||
return
|
||||
for key, child in node.children.items():
|
||||
if child.terminates:
|
||||
result.append(curr_word+key)
|
||||
self._list_words(child, curr_word+key, result)
|
||||
result.append(curr_word + key)
|
||||
self._list_words(child, curr_word + key, result)
|
||||
|
||||
@@ -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
|
||||
}
|
||||
|
||||
@@ -165,9 +165,7 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 1,
|
||||
"metadata": {
|
||||
"collapsed": false
|
||||
},
|
||||
"metadata": {},
|
||||
"outputs": [
|
||||
{
|
||||
"name": "stdout",
|
||||
@@ -260,9 +258,7 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 2,
|
||||
"metadata": {
|
||||
"collapsed": false
|
||||
},
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"%run trie.py"
|
||||
@@ -278,9 +274,7 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 3,
|
||||
"metadata": {
|
||||
"collapsed": false
|
||||
},
|
||||
"metadata": {},
|
||||
"outputs": [
|
||||
{
|
||||
"name": "stdout",
|
||||
@@ -292,11 +286,10 @@
|
||||
],
|
||||
"source": [
|
||||
"%%writefile test_trie.py\n",
|
||||
"from nose.tools import assert_true\n",
|
||||
"from nose.tools import raises\n",
|
||||
"import unittest\n",
|
||||
"\n",
|
||||
"\n",
|
||||
"class TestTrie(object): \n",
|
||||
"class TestTrie(unittest.TestCase): \n",
|
||||
"\n",
|
||||
" def test_trie(self):\n",
|
||||
" trie = Trie()\n",
|
||||
@@ -307,7 +300,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",
|
||||
" self.assertTrue(trie.find(word) is not None)\n",
|
||||
" \n",
|
||||
" print('Test: Remove me')\n",
|
||||
" trie.remove('me')\n",
|
||||
@@ -315,9 +308,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",
|
||||
@@ -325,9 +318,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",
|
||||
@@ -335,9 +328,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",
|
||||
@@ -345,23 +338,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",
|
||||
" @raises(Exception)\n",
|
||||
" def test_trie_remove_invalid(self):\n",
|
||||
" print('Test: Remove from empty trie')\n",
|
||||
" trie = Trie()\n",
|
||||
" assert_true(trie.remove('foo') is None) \n",
|
||||
" self.assertTrue(trie.remove('foo') is None) \n",
|
||||
"\n",
|
||||
"\n",
|
||||
"def main():\n",
|
||||
" test = TestTrie()\n",
|
||||
" test.test_trie()\n",
|
||||
" test.test_trie_remove_invalid()\n",
|
||||
" test.assertRaises(KeyError, test.test_trie_remove_invalid)\n",
|
||||
"\n",
|
||||
"\n",
|
||||
"if __name__ == '__main__':\n",
|
||||
@@ -371,9 +363,7 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 4,
|
||||
"metadata": {
|
||||
"collapsed": false
|
||||
},
|
||||
"metadata": {},
|
||||
"outputs": [
|
||||
{
|
||||
"name": "stdout",
|
||||
@@ -410,9 +400,9 @@
|
||||
"name": "python",
|
||||
"nbconvert_exporter": "python",
|
||||
"pygments_lexer": "ipython3",
|
||||
"version": "3.4.3"
|
||||
"version": "3.7.2"
|
||||
}
|
||||
},
|
||||
"nbformat": 4,
|
||||
"nbformat_minor": 0
|
||||
"nbformat_minor": 1
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user