Beispiel #1
0
    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)
Beispiel #2
0
    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)