class TestFindWords(TestCase): """Tests for the find_words method.""" def setUp(self): self._word_finder = WordFinder() def test_find_words(self): # Test find_words by comparing its output against a slower # but quite surely correct implementation of checking all the # permutations against the words database. s = "anbheiiqzmlp" r = 5 gold_set = set() for perm in permutations(s, r): word = "".join(perm) if word in self._word_finder._word_db: gold_set.add(word) test_set = set(self._word_finder.find_words(s, r)) self.assertSetEqual(gold_set, test_set)
def setUp(self): self._word_finder = WordFinder()