def test_contains(self): c = PrefixCache(Word, 'normalized_spelling') c.seed([u'ab-starting-words']) self.assertEqual(c.items(), [(u'ab', set([u'abc']))]) self.assertTrue(c.contains(u'abc')) self.assertTrue(c.contains(u'efg')) self.assertEqual(c.items(), [(u'ab', set([u'abc'])), (u'ef', set([u'efg']))])
def test_seed(self): c = PrefixCache(Word, 'normalized_spelling') c.seed([u'abx', u'efh']) self.assertEqual(sorted(c.items()), [(u'ab', set([u'abc'])), (u'ef', set([u'efg']))])
def test_seed_with_no_values(self): c = PrefixCache(Word, 'normalized_spelling') c.seed([]) self.assertEqual(c.items(), [])