add test_mecab_system_eval to nagisa_test.py

taishi-i · Jun 11, 2020 · d793068 · d793068
1 parent d9fa1cf
commit d793068
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/test/nagisa_test.py b/test/nagisa_test.py
@@ -121,6 +121,18 @@ def test_fit(self):
         )
 
 
+    def test_mecab_system_eval(self):
+        # test_19
+        system_file = "nagisa/data/sample_datasets/sample.train"
+        answer_file = "nagisa/data/sample_datasets/sample.train"
+
+        system_data = nagisa.train.mecab_system_eval.readFile(system_file)
+        answer_data = nagisa.train.mecab_system_eval.readFile(answer_file)
+
+        r = nagisa.train.mecab_system_eval.mecab_eval(system_data, answer_data)
+        nagisa.train.mecab_system_eval.print_eval(r)
+
+
 def suite():
     suite = unittest.TestSuite()
     suite.addTests(unittest.makeSuite(TestNagisa))