fix test_mecab_system_eval

taishi-i · Jun 11, 2020 · ed07efb · ed07efb
1 parent 9a1a115
commit ed07efb
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 2 deletions.
diff --git a/nagisa/data/sample_datasets/sample.pred b/nagisa/data/sample_datasets/sample.pred
@@ -0,0 +1,29 @@
+福岡	名詞
+・	補助記号
+博	名詞
+多	名詞
+の	助詞
+観光	名詞
+情報	名詞
+EOS
+Pyt	名詞
+hon	名詞
+で	助詞
+簡単	形状詞
+に	助動詞
+使える	動詞
+ツール	名詞
+です	助動詞
+EOS
+顔	名詞
+文	名詞
+字	名詞
+や	助詞
+URL	名詞
+に	助詞
+対し	動詞
+て	助詞
+頑健	名詞
+な	助動詞
+解析	名詞
+EOS
diff --git a/test/nagisa_test.py b/test/nagisa_test.py
@@ -123,14 +123,17 @@ 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_file = "nagisa/data/sample_datasets/sample.pred"
+        answer_file = "nagisa/data/sample_datasets/sample.test"
 
         system_data = nagisa.train.mecab_system_eval.readFile(system_file)
         answer_data = nagisa.train.mecab_system_eval.readFile(answer_file)
 
+        expected_r = [20, 20, 26, 23]
         r = nagisa.train.mecab_system_eval.mecab_eval(system_data, answer_data)
+
         nagisa.train.mecab_system_eval.print_eval(r)
+        self.assertEqual(r, expected_r)
 
 
 def suite():