Use xps.real_dtypes() in tests for comparison functions

Author: honno

array_api_tests/test_searching_functions.py

@@ -16,7 +16,7 @@
 
 @given(
     x=xps.arrays(
-        dtype=xps.numeric_dtypes(),
+        dtype=xps.real_dtypes(),
         shape=hh.shapes(min_dims=1, min_side=1),
         elements={"allow_nan": False},
     ),
@@ -53,7 +53,7 @@ def test_argmax(x, data):
 
 @given(
     x=xps.arrays(
-        dtype=xps.numeric_dtypes(),
+        dtype=xps.real_dtypes(),
         shape=hh.shapes(min_dims=1, min_side=1),
         elements={"allow_nan": False},
     ),

array_api_tests/test_sorting_functions.py

@@ -34,7 +34,7 @@ def assert_scalar_in_set(
 # TODO: Test with signed zeros and NaNs (and ignore them somehow)
 @given(
     x=xps.arrays(
-        dtype=xps.scalar_dtypes(),
+        dtype=xps.real_dtypes(),
         shape=hh.shapes(min_dims=1, min_side=1),
         elements={"allow_nan": False},
     ),
@@ -94,7 +94,7 @@ def test_argsort(x, data):
 # TODO: Test with signed zeros and NaNs (and ignore them somehow)
 @given(
     x=xps.arrays(
-        dtype=xps.scalar_dtypes(),
+        dtype=xps.real_dtypes(),
         shape=hh.shapes(min_dims=1, min_side=1),
         elements={"allow_nan": False},
     ),

array_api_tests/test_statistical_functions.py

@@ -27,7 +27,7 @@ def kwarg_dtypes(dtype: DataType) -> st.SearchStrategy[Optional[DataType]]:
 
 @given(
     x=xps.arrays(
-        dtype=xps.numeric_dtypes(),
+        dtype=xps.real_dtypes(),
         shape=hh.shapes(min_side=1),
         elements={"allow_nan": False},
     ),
@@ -79,7 +79,7 @@ def test_mean(x, data):
 
 @given(
     x=xps.arrays(
-        dtype=xps.numeric_dtypes(),
+        dtype=xps.real_dtypes(),
         shape=hh.shapes(min_side=1),
         elements={"allow_nan": False},
     ),