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)
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)