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