Add additional test cases for NewTrie

tree/trie/test_new_trie.py

@@ -26,5 +26,35 @@ def test_non_existent_word(self):
         self.trie.insert("hello")
         self.assertFalse(self.trie.search("world"))
 
+    def test_words_with_common_prefixes(self):
+        words = ["tree", "trie", "try", "triangle", "trap"]
+        for word in words:
+            self.trie.insert(word)
+        for word in words:
+            self.assertTrue(self.trie.search(word))
+        self.assertFalse(self.trie.search("tr"))
+        self.assertFalse(self.trie.search("tria"))
+
+    def test_prefix_search(self):
+        self.trie.insert("programming")
+        self.assertTrue(self.trie.search("programming"))
+        self.assertFalse(self.trie.search("program"))
+        self.assertFalse(self.trie.search("prog"))
+
+    def test_single_character_words(self):
+        for char in "abcdefghijklmnopqrstuvwxyz":
+            self.trie.insert(char)
+        for char in "abcdefghijklmnopqrstuvwxyz":
+            self.assertTrue(self.trie.search(char))
+        self.assertFalse(self.trie.search(""))
+
+    def test_similar_non_existent_words(self):
+        self.trie.insert("hello")
+        self.trie.insert("world")
+        self.assertFalse(self.trie.search("hell"))
+        self.assertFalse(self.trie.search("worl"))
+        self.assertFalse(self.trie.search("he"))
+        self.assertFalse(self.trie.search("wo"))
+
 if __name__ == "__main__":
     unittest.main()