def test_basic(self):
trie = Trie()
trie.insert("cat")
self.assertEqual(trie.__len__(), 1)
self.assertTrue(trie.__contains__("cat"))
self.assertEqual([x for x in trie.__iter__()], ["cat"])
self.assertTrue(trie.contains_prefix("ca"))
self.assertEqual([x for x in trie.prefix_iter("ca")], ["cat"])
class TrieTester(unittest.TestCase):
def setUp(self):
self.Trie = Trie()
def test_simple_insert(self):
self.Trie.insert("hel")
self.Trie.insert("hello")
self.assertTrue("hello" in self.Trie)
def test_contains_prefix(self):
self.Trie.insert("Hello")
self.assertTrue(self.Trie.contains_prefix("Hel"))
"""
