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")) """