def test_setup_sets_table(self): sets = anagram.setup_sets_table(self.hashtable) self.assertEqual(len(sets), 2) keys = sets.keys() self.assertEqual(keys[0], 2) self.assertEqual(keys[1], 4) values = sets.values() self.assertEqual(len(values[0][0]), 2) self.assertEqual(len(values[1][0]), 4)
def test_print_sets(self): sets = anagram.setup_sets_table(self.hashtable) anagram.print_sets(sets) with open('out.txt', 'r') as file: lists = list(filter(None, (line.rstrip() for line in file))) largest_set = lists[0] self.assertTrue('team' in largest_set) self.assertTrue('meat' in largest_set) self.assertTrue('mate' in largest_set) self.assertTrue('tame' in largest_set) self.assertTrue('hello' in lists[1]) self.assertTrue('ellho' in lists[1])