def test_calc_relative_word_frequency_word_in_both_documents(self):
words = {
"alpha": 10,
}
rel_word_frequency = WordCloud.calc_relative_word_freq(
Counter(words), Counter({"alpha": 9}))
self.assertEqual({"alpha": 0.5}, rel_word_frequency)
def test_calc_relative_word_frequency_word_in_contrast(self):
words = {
"alpha": 10,
}
rel_word_frequency = WordCloud.calc_relative_word_freq(
Counter(words), Counter({"alpha": 4}))
self.assertEqual({"alpha": 2}, rel_word_frequency)
def test_calc_relative_word_frequency_word_in_normal_word_freq(self):
words = {
"alpha": 10,
}
rel_word_frequency = WordCloud.calc_relative_word_freq(
Counter(words), Counter())
self.assertEqual({"alpha": 5}, rel_word_frequency)
def test_calc_relative_word_frequency_base_case(self):
words = {"alpha": 10, "bravo": 5}
rel_word_frequency = WordCloud.calc_relative_word_freq(
Counter(words), Counter())
self.assertEqual(words, rel_word_frequency)
