def generate_word_clouds(self): save_words = self.get_word_frequency_for_viewpoint(False) repeal_words = self.get_word_frequency_for_viewpoint(True) save_cloud = WordCloud(save_words, repeal_words) self.add_to_cache("vn:cloud", save_cloud.as_dict()) repeal_cloud = WordCloud(repeal_words, save_words) self.add_to_cache("vp:cloud", repeal_cloud.as_dict())
def test_as_dict(self): wc = WordCloud(Counter({ "alpha": 10, "beta": 5 }), Counter({"alpha": 4})) self.assertEqual({"beta": 5, "alpha": 2}, wc.as_dict())