def test_rouge_n1_all(self): self.assertAlmostEqual(rouge_n(self.s1, self.references, 1), 0.25, 2)
def test_rouge_n2_all(self): self.assertAlmostEqual(rouge_n(self.s1, self.references, 2), 0.17, 2)
def test_rouge_n3(self): self.assertEqual(rouge_n(self.s1, [self.s2], 3), 0.0)
def test_rouge_n2(self): self.assertAlmostEqual(rouge_n(self.s1, [self.s2], 2), 0.33, 2)
def test_rouge_n1(self): self.assertAlmostEqual(rouge_n(self.s1, [self.s2], 1), 0.75, 2)
def test_rouge_n_identiy_2(self): # if n is so large that no ngram fits, # then ROUGE-N is zero.... ! self.assertEqual(rouge_n(self.s1, [self.s1], len(self.s1) + 1), 0.0)
def test_rouge_n_identiy_1(self): for n in range(1, len(self.s1) + 1): self.assertEqual(rouge_n(self.s1, [self.s1], n), 1.0)