Python Rake._build_word_co_occurance_graph Beispiele

Beispiel #1

Datei: rake_test.py Projekt: kabirkhan/rake-nltk

    def test_build_word_co_occurance_graph(self):
        r = Rake()

        phrase_list = [['red', 'apples'], ['good'], ['red'], ['flavour']]
        degree = defaultdict(lambda: 0)
        degree['apples'] = 2
        degree['good'] = 1
        degree['flavour'] = 1
        degree['red'] = 3
        r._build_word_co_occurance_graph(phrase_list)
        self.assertEqual(r.get_word_degrees(), degree)

Beispiel #2

    def test_build_word_co_occurance_graph(self):
        r = Rake()

        phrase_list = [["red", "apples"], ["good"], ["red"], ["flavour"]]
        degree = defaultdict(lambda: 0)
        degree["apples"] = 2
        degree["good"] = 1
        degree["flavour"] = 1
        degree["red"] = 3
        r._build_word_co_occurance_graph(phrase_list)
        self.assertEqual(r.get_word_degrees(), degree)