More dtypes

xarray-contrib · Apr 1, 2024 · 82db283 · 82db283
1 parent 28db1dc
commit 82db283
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/tests/test_properties.py b/tests/test_properties.py
@@ -21,17 +21,15 @@ def supported_dtypes() -> st.SearchStrategy[np.dtype]:
         npst.integer_dtypes()
         | npst.unsigned_integer_dtypes()
         | npst.floating_dtypes()
-        # | npst.complex_number_dtypes()
-        # | npst.datetime64_dtypes()
-        # | npst.timedelta64_dtypes()
+        | npst.complex_number_dtypes()
+        | npst.datetime64_dtypes()
+        | npst.timedelta64_dtypes()
         | npst.unicode_string_dtypes()
-    )
+    ).filter(lambda x: x.byteorder == "=")
 
 
-dtype_st = supported_dtypes().filter(lambda x: x.byteorder == "=" and x.kind not in ["mcM"])
-
-array_dtype_st = dtype_st.filter(lambda x: x.kind != "U")
-by_dtype_st = dtype_st
+array_dtype_st = supported_dtypes().filter(lambda x: x.kind != "mcMU")
+by_dtype_st = supported_dtypes()
 func_st = st.sampled_from([f for f in ALL_FUNCS if f not in NON_NUMPY_FUNCS])