def test_ignore_case_predict(self): if sys.version_info.major < 3: return trie = Trie(ignore_case=True, ordered=True) ids = {w: trie.insert(w) for w in [u"aaİ", u"aİİ", u"aai̇", u"aai̇bİ"]} predicts = list(trie.predict(u"aaİ")) self.assertEqual(predicts, [ids[u'aai̇'], ids[u'aai̇bİ']])
def test_predict(self): trie = Trie(ordered=True) ids = { w: trie.insert(w) for w in [u"ruby", u"rubx", u"rab", u"rub", u"rb"] } predicts = list(trie.predict(u"r")) self.assertEqual( predicts, [ids[u"rb"], ids[u"rab"], ids[u"rub"], ids[u"rubx"], ids[u"ruby"]])