mirror of
https://github.com/moses-smt/mosesdecoder.git
synced 2024-10-27 11:59:52 +03:00
29 lines
709 B
Python
Executable File
29 lines
709 B
Python
Executable File
#!/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)
|
|
|