Fix dtypes tests to work with numpy 1 or 2

PyO3 · Sep 24, 2024 · 6013343 · 6013343
1 parent da0807f
commit 6013343
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/dtype.rs b/src/dtype.rs
@@ -826,7 +826,7 @@ mod tests {
 
     use pyo3::{py_run, types::PyTypeMethods};
 
-    use crate::npyffi::NPY_NEEDS_PYAPI;
+    use crate::npyffi::{is_numpy_2, NPY_NEEDS_PYAPI};
 
     #[test]
     fn test_dtype_new() {
@@ -855,7 +855,11 @@ mod tests {
             dtype_bound::<T>(py).typeobj().qualname().unwrap()
         }
         Python::with_gil(|py| {
-            assert_eq!(type_name::<bool>(py), "bool_");
+            if is_numpy_2(py) {
+                assert_eq!(type_name::<bool>(py), "bool");
+            } else {
+                assert_eq!(type_name::<bool>(py), "bool_");
+            }
 
             assert_eq!(type_name::<i8>(py), "int8");
             assert_eq!(type_name::<i16>(py), "int16");