Fix some test cases

Why do they change?
huggingface · Mar 22, 2022 · f5276d8 · f5276d8
1 parent e49a768
commit f5276d8
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git a/bindings/python/src/decoders.rs b/bindings/python/src/decoders.rs
@@ -420,7 +420,7 @@ mod test {
         let py_meta = py_dec.get_as_subtype().unwrap();
         let gil = Python::acquire_gil();
         assert_eq!(
-            "tokenizers.decoders.Metaspace",
+            "Metaspace",
             py_meta.as_ref(gil.python()).get_type().name().unwrap()
         );
     }

diff --git a/bindings/python/src/models.rs b/bindings/python/src/models.rs
@@ -811,7 +811,7 @@ mod test {
         let py_bpe = py_model.get_as_subtype().unwrap();
         let gil = Python::acquire_gil();
         assert_eq!(
-            "tokenizers.models.BPE",
+            "BPE",
             py_bpe.as_ref(gil.python()).get_type().name().unwrap()
         );
     }

diff --git a/bindings/python/src/normalizers.rs b/bindings/python/src/normalizers.rs
@@ -628,7 +628,7 @@ mod test {
         let py_nfc = py_norm.get_as_subtype().unwrap();
         let gil = Python::acquire_gil();
         assert_eq!(
-            "tokenizers.normalizers.NFC",
+            "NFC",
             py_nfc.as_ref(gil.python()).get_type().name().unwrap()
         );
     }

diff --git a/bindings/python/src/pre_tokenizers.rs b/bindings/python/src/pre_tokenizers.rs
@@ -706,7 +706,7 @@ mod test {
         let py_wsp = py_norm.get_as_subtype().unwrap();
         let gil = Python::acquire_gil();
         assert_eq!(
-            "tokenizers.pre_tokenizers.Whitespace",
+            "Whitespace",
             py_wsp.as_ref(gil.python()).get_type().name().unwrap()
         );
     }

diff --git a/bindings/python/src/processors.rs b/bindings/python/src/processors.rs
@@ -433,7 +433,7 @@ mod test {
         let py_bert = py_proc.get_as_subtype().unwrap();
         let gil = Python::acquire_gil();
         assert_eq!(
-            "tokenizers.processors.BertProcessing",
+            "BertProcessing",
             py_bert.as_ref(gil.python()).get_type().name().unwrap()
         );
     }

diff --git a/bindings/python/src/trainers.rs b/bindings/python/src/trainers.rs
@@ -868,7 +868,7 @@ mod tests {
         let py_bpe = py_trainer.get_as_subtype().unwrap();
         let gil = Python::acquire_gil();
         assert_eq!(
-            "tokenizers.trainers.BpeTrainer",
+            "BpeTrainer",
             py_bpe.as_ref(gil.python()).get_type().name().unwrap()
         );
     }