def test_build_frequency_dist(self): r = Rake() phrase_list = [['red', 'apples'], ['good'], ['red'], ['flavour']] freq = defaultdict(lambda: 0) freq['apples'] = 1 freq['good'] = 1 freq['flavour'] = 1 freq['red'] = 2 r._build_frequency_dist(phrase_list) self.assertEqual(r.get_word_frequency_distribution(), freq)
def test_build_frequency_dist(self): r = Rake() phrase_list = [["red", "apples"], ["good"], ["red"], ["flavour"]] freq = defaultdict(lambda: 0) freq["apples"] = 1 freq["good"] = 1 freq["flavour"] = 1 freq["red"] = 2 r._build_frequency_dist(phrase_list) self.assertEqual(r.get_word_frequency_distribution(), freq)