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)