Also test on a suffix style input

himkt · Dec 26, 2019 · ce3d581 · ce3d581
1 parent e9cd6c4
commit ce3d581
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/test_metrics.py b/tests/test_metrics.py
@@ -43,7 +43,11 @@ def test_get_entities_with_unexpected_input(self):
         y_true = ['O', 'O', 'O', 'MISC', 'MISC', 'MISC', 'O', 'PER', 'PER']
         with self.assertRaises(Exception):
             get_entities(y_true)
-            get_entities(y_true, suffix=True)
+
+    def test_get_entities_with_unexpected_input_and_suffix_style(self):
+        y_true = ['O', 'O', 'O', 'MISC', 'MISC', 'MISC', 'O', 'PER', 'PER']
+        with self.assertRaises(Exception):
+            get_entities(y_true)
 
     def test_performance_measure(self):
         y_true = [['O', 'O', 'O', 'B-MISC', 'I-MISC', 'O', 'B-ORG'], ['B-PER', 'I-PER', 'O']]