TST: fix testing on Windows against numpy2

yt-project · Jul 1, 2024 · 8093a92 · 8093a92
1 parent 93ceb32
commit 8093a92
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/ewah_bool_utils/_testing.pyx b/ewah_bool_utils/_testing.pyx
@@ -15,6 +15,10 @@ ctypedef fused dtype_t:
     np.float32_t
     np.float64_t
 
+ctypedef fused int_t:
+    np.int32_t
+    np.int64_t
+
 
 cdef class Index:
     cdef void* ewah_array
@@ -73,7 +77,7 @@ cdef class Index:
             ids.size, ewah_array[0].sizeInBytes()))
 
 
-cpdef np.uint64_t[:] ewah_set_and_unset(long[:] arr):
+cpdef np.uint64_t[:] ewah_set_and_unset(np.ndarray[int_t, ndim=1] arr):
     cdef ewah_bool_array *ewah_arr
     cdef vector[size_t] vec