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