File size: 618 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 |
#!/usr/bin/env python
import unittest
from main import *
class TestConfig(unittest.TestCase):
def setUp(self):
self.config = Config()
def test_files_short(self):
nbests = ["nbest1", "nbest2", "nbest3"]
scores = ["score1", "score2", "score3"]
self.config.parse(["-n", nbests[0], "-n", nbests[1], "-n", nbests[2],\
"-S", scores[0], "-S", scores[1], "-S", scores[2]])
self.assertEqual(self.config.nbest_files, nbests)
self.assertEqual(self.config.score_files, scores)
if __name__ == "__main__":
unittest.main()
suite = unittest.TestLoader().loadTestsFromTestCase(TestConfig)
|