def test_ignore_case_prefix(self):
if sys.version_info.major < 3:
return
txt = u"aaİbİc"
trie = Trie(ignore_case=True)
ids = {w: trie.insert(w) for w in [u"aİ", u"aİİ", u"aai̇", u"aai̇bİ"]}
prefixes = list(trie.prefix(txt))
self.assertEqual(prefixes, [(ids[u"aai̇"], 3), (ids[u"aai̇bİ"], 5)])
txt = u"aai̇bİc"
prefixes = list(trie.prefix(txt))
self.assertEqual(prefixes, [(ids[u"aai̇"], 4), (ids[u"aai̇bİ"], 6)])
def test_prefix(self):
trie = Trie()
ids = {w: trie.insert(w) for w in [u"ruby", u"rubx", u"rab", u"rub"]}
prefixes = list(trie.prefix(u"ruby on rails"))
self.assertEqual(prefixes, [(ids[u"rub"], 3), (ids[u"ruby"], 4)])