示例#1
0
    def test_get_top_words_over_100_words(self):
        data = {str(i): i for i in range(0, 150)}
        words = WordCloud.get_top_words(data)

        self.assertEqual(100, len(words))

        expected_top_words = [(str(i), i) for i in range(149, 49, -1)]
        self.assertEqual(expected_top_words, words)
示例#2
0
 def test_get_top_words_normal_case(self):
     words = WordCloud.get_top_words({"A": 1, "B": 3, "C": 2})
     self.assertEqual([("B", 3), ("C", 2), ("A", 1)], words)