def test_discard(self): c = PrefixCache(Word, 'normalized_spelling') c.add(u'jkl') c.add(u'mno') c.add(u'mnp') self.assertEqual(sorted(c.items()), [(u'jk', set([u'jkl'])), (u'mn', set([u'mno', u'mnp']))]) c.discard(u'jkl') c.discard(u'mno') self.assertEqual(c.items(), [(u'mn', set([u'mnp'])), (u'jk', set([]))])
def test_add(self): c = PrefixCache(Word, 'normalized_spelling') c.add(u'ghi') self.assertEqual(c.items(), [(u'gh', set([u'ghi']))])