Fixed test_full_like

Author: honno

array_api_tests/hypothesis_helpers.py

@@ -2,11 +2,11 @@
 from operator import mul
 from math import sqrt
 
+from hypothesis import assume
 from hypothesis.strategies import (lists, integers, sampled_from,
                                    shared, floats, just, composite, one_of,
                                    none, booleans)
 from hypothesis.extra.array_api import make_strategies_namespace
-from hypothesis import assume
 
 from .pytest_helpers import nargs
 from .array_helpers import (dtype_ranges, integer_dtype_objects,

array_api_tests/meta_tests/test_hypothesis_helpers.py

@@ -1,7 +1,7 @@
 from math import prod
 
 import pytest
-from hypothesis import given, strategies as st, assume
+from hypothesis import given, strategies as st
 
 from .. import _array_module as xp
 from .._array_module import _UndefinedStub
@@ -70,4 +70,3 @@ def run(kw):
     c_results = [kw for kw in results if "c" in kw]
     assert len(c_results) > 0
     assert all(isinstance(kw["c"], str) for kw in c_results)
-

array_api_tests/test_creation_functions.py

@@ -8,7 +8,7 @@
                                  scalars, xps, kwargs)
 
 from hypothesis import assume, given
-from hypothesis.strategies import integers, floats, one_of, none, booleans, just, shared
+from hypothesis.strategies import integers, floats, one_of, none, booleans, just, shared, composite
 
 
 optional_dtypes = none() | shared_dtypes
@@ -148,10 +148,17 @@ def test_full(shape, fill_value, dtype):
         assert all(equal(a, asarray(fill_value, **kwargs))), "full() array did not equal the fill value"
 
 
+@composite
+def fill_values(draw):
+    kw = draw(shared(kwargs(dtype=none() | xps.scalar_dtypes()), key="full_like_kw"))
+    dtype = kw.get("dtype", None) or draw(shared_dtypes)
+    return draw(xps.from_dtype(dtype))
+
+
 @given(
     x=xps.arrays(dtype=shared_dtypes, shape=shapes),
-    fill_value=shared_dtypes.flatmap(xps.from_dtype),
-    kw=kwargs(dtype=none() | shared_dtypes),
+    fill_value=fill_values(),
+    kw=shared(kwargs(dtype=none() | xps.scalar_dtypes()), key="full_like_kw"),
 )
 def test_full_like(x, fill_value, kw):
     out = full_like(x, fill_value, **kw)