File size: 709 Bytes
158b61b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
#!/usr/bin/env python
import math
import unittest
from bleu import *
class TestBleuScorer(unittest.TestCase):
def setUp(self):
self.scorer = BleuScorer()
def test_wrong_length(self):
self.assertRaises(BleuScoreException, self.scorer.score, [1]*8)
self.assertRaises(BleuScoreException, self.scorer.score, [1]*10)
def test_score(self):
stats = [2,5,0,1,1,1,1,1,5]
self.assertAlmostEqual(self.scorer.score(stats), 1/math.sqrt(2))
def test_brevity(self):
stats = [2,2,2,2,2,2,2,2,3]
self.assertAlmostEqual(self.scorer.score(stats), math.exp(1 - 3.0/2.0))
if __name__ == "__main__":
unittest.main()
suite = unittest.TestLoader().loadTestsFromTestCase(TestBleuScorer)
|