def test_filter_words_both_none(self): """ Frequencies dict is none """ expected_result = {} freq_dict = expected_result main.filter_stop_words(None, None) self.assertEqual(expected_result, freq_dict)
def test_filter_words_stop_numbers(self): """ Stop words with numbers """ freq_dict = { 'the': 2, 'quick': 1, 'brown': 1, 'fox': 1, 'jumps': 1, 'over': 1, 'lazy': 1, 'dog': 1 } expected_result = { 'quick': 1, 'brown': 1, 'fox': 1, 'jumps': 1, 'lazy': 1, 'dog': 1 } res = main.filter_stop_words(freq_dict, ('the', 1, 'over', 4)) self.assertEqual(expected_result, res)
def test_filter_words_ideal(self): """ Filter stop words ideal case - good dict, good stop words """ freq_dict = { 'the': 2, 'quick': 1, 'brown': 1, 'fox': 1, 'jumps': 1, 'over': 1, 'lazy': 1, 'dog': 1 } expected_result = { 'quick': 1, 'brown': 1, 'fox': 1, 'jumps': 1, 'lazy': 1, 'dog': 1 } res = main.filter_stop_words(freq_dict, STOP_WORDS) self.assertEqual(expected_result, res)
def test_filter_words__dict_no_str(self): """ Filter stop words when dict has keys - not strs """ expected_result = {'fox': 2} res = main.filter_stop_words({1: 1, 'fox': 2}, STOP_WORDS) self.assertEqual(expected_result, res)
def test_filter_words_dict_empty(self): """ Filter stop words when stop words tuple is empty """ expected_result = {} res = main.filter_stop_words({}, STOP_WORDS) self.assertEqual(expected_result, res)
def test_filter_words_stop_none(self): """ Stop words is none """ expected_result = { 'the': 2, 'quick': 1, 'brown': 1, 'fox': 1, 'jumps': 1, 'over': 1, 'lazy': 1, 'dog': 1 } freq_dict = expected_result main.filter_stop_words(freq_dict, None) self.assertEqual(expected_result, freq_dict)
def test_filter_words_copy(self): """ Check that frequencies dict is immutable """ expected_result = { 'the': 2, 'quick': 1, 'brown': 1, 'fox': 1, 'jumps': 1, 'over': 1, 'lazy': 1, 'dog': 1 } freq_dict = expected_result main.filter_stop_words(freq_dict, STOP_WORDS) self.assertEqual(expected_result, freq_dict)
def test_filter_words_empty(self): """ Filter stop words when stop words tuple is empty """ expected_result = { 'the': 2, 'quick': 1, 'brown': 1, 'fox': 1, 'jumps': 1, 'over': 1, 'lazy': 1, 'dog': 1 } res = main.filter_stop_words(expected_result, ()) self.assertEqual(expected_result, res)