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