remove int16 but test the float dtypes

(which only works by restricting the values)
IAOCEA · Jul 25, 2024 · 7f5ec6e · 7f5ec6e
1 parent c726356
commit 7f5ec6e
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/healpix_convolution/tests/test_convolution.py b/healpix_convolution/tests/test_convolution.py
@@ -11,7 +11,13 @@
     data=npst.arrays(
         shape=st.sampled_from([(5,), (10, 5)]),
         # TODO: figure out how to deal with floating point values
-        dtype=st.sampled_from(["int16", "int32", "int64"]),
+        dtype=st.sampled_from(["int32", "int64", "float32", "float64"]),
+        elements={
+            "allow_infinity": False,
+            "allow_subnormal": False,
+            "min_value": 1e9,
+            "max_value": 1e9,
+        },
     ),
 )
 @settings(deadline=1000)