fix test

eole-nlp · Oct 18, 2024 · 1630114 · 1630114
1 parent da7689d
commit 1630114
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/eole/tests/test_transform.py b/eole/tests/test_transform.py
@@ -338,8 +338,7 @@ def test_sentencepiece(self):
                 "s",
                 "p",
                 "on",
-                "s",
-                "e",
+                "se",
             ],
             "tgt": [],
         }
@@ -412,7 +411,7 @@ def test_pyonmttok_bpe(self):
                 "on￭",
                 "se",
                 ":",
-                "｟newline｠",
+                "｟newline｠￭",
                 "respon￭",
                 "se",
             ],
@@ -813,4 +812,4 @@ def test_insert_mask_before_placeholder(self):
                 "｟newline｠response",
             ],
         }
-        self.assertEqual(ex_out, ex_gold)
+        self.assertEqual(ex_in, ex_gold)