def test_remove_substrings(self): """ Doesn't include substring words in the output dictionary """ test_counts = {'corgi': 10, # score 50 'corgicat': 5} # score 40 expected_dict = 'corgicat' d = zdict_gen.genDictFromFreq(test_counts, -1) self.assertEqual(d, expected_dict)
def test_remove_substrings(self): """ Doesn't include substring words in the output dictionary """ test_counts = { 'corgi': 10, # score 50 'corgicat': 5 } # score 40 expected_dict = 'corgicat' d = zdict_gen.genDictFromFreq(test_counts, -1) self.assertEqual(d, expected_dict)
def test_is_corrrect_order(self): """ Should correctly score and generate dictionary """ test_counts = {'corgi': 10, # score 50 'cat': 5, # score 15 'lizard': 6, # score 36 'gecko': 7, # score 35 'hamster': 4, # score 28 'parrot': 3} # score 18 full_dict = zdict_gen.genDictFromFreq(test_counts, -1) expected_dict = 'catparrothamstergeckolizardcorgi' self.assertEqual(full_dict, expected_dict)
def test_is_corrrect_order(self): """ Should correctly score and generate dictionary """ test_counts = { 'corgi': 10, # score 50 'cat': 5, # score 15 'lizard': 6, # score 36 'gecko': 7, # score 35 'hamster': 4, # score 28 'parrot': 3 } # score 18 full_dict = zdict_gen.genDictFromFreq(test_counts, -1) expected_dict = 'catparrothamstergeckolizardcorgi' self.assertEqual(full_dict, expected_dict)
def test_equal_to_size_b(self): """ Output of dictionary should be equal to size_b if size_b is valid and the unlimited dictionary would have been larger than size_b """ test_counts = {'corgi': 10, # score 50 'cat': 5, # score 15 'lizard': 6, # score 36 'gecko': 7, # score 35 'hamster': 4, # score 28 'parrot': 3} # score 18 expected_dict = 'geckolizardcorgi' expected_size = len(expected_dict) small_dict = zdict_gen.genDictFromFreq(test_counts, expected_size) self.assertEqual(small_dict, expected_dict)
def test_equal_to_size_b(self): """ Output of dictionary should be equal to size_b if size_b is valid and the unlimited dictionary would have been larger than size_b """ test_counts = { 'corgi': 10, # score 50 'cat': 5, # score 15 'lizard': 6, # score 36 'gecko': 7, # score 35 'hamster': 4, # score 28 'parrot': 3 } # score 18 expected_dict = 'geckolizardcorgi' expected_size = len(expected_dict) small_dict = zdict_gen.genDictFromFreq(test_counts, expected_size) self.assertEqual(small_dict, expected_dict)