def test_calc_ROUGE4(self): N = 4 TPs = [1, 1, 1, 1] FNs = [1, 1, 1, 1] self.assertAlmostEqual(eval.calc_ROUGE(N, TPs, FNs), (4/27)**(1/4))
def test_calc_ROUGE5(self): N = 4 TPs = [30, 20, 0, 0] FNs = [20, 10, 1, 0] self.assertAlmostEqual(eval.calc_ROUGE(N, TPs, FNs), (3*21/(5*31*2*2))**(1/4))
def test_calc_ROUGE3(self): N = 4 TPs = [0, 0, 0, 0] FNs = [1, 1, 1, 1] self.assertEqual(eval.calc_ROUGE(N, TPs, FNs), 0)