Skip to content

Missing use cases for binops on array vs. scalar #364

Open
@crusaderky

Description

@crusaderky

After reading hh.scalars, I understand that array-api-tests currently never generates the use cases of a binop where one argument is an Array with float or complex dtype and the other argument is

  • a pure-python integer;
  • a scalar inf, -inf, or nan;
  • a scalar outside of the range [1/64, 64]

Unless I missed something and the tests for these cases are somewhere else?

@composite
def scalars(draw, dtypes, finite=False, **kwds):
"""
Strategy to generate a scalar that matches a dtype strategy
dtypes should be one of the shared_* dtypes strategies.
"""
dtype = draw(dtypes)
if dh.is_int_dtype(dtype):
m, M = dh.dtype_ranges[dtype]
return draw(integers(m, M))
elif dtype == bool_dtype:
return draw(booleans())
elif dtype == float64:
if finite:
return draw(floats(allow_nan=False, allow_infinity=False, **kwds))
return draw(floats(), **kwds)
elif dtype == float32:
if finite:
return draw(floats(width=32, allow_nan=False, allow_infinity=False, **kwds))
return draw(floats(width=32, **kwds))
elif dtype == complex64:
if finite:
return draw(complex_numbers(width=32, allow_nan=False, allow_infinity=False))
return draw(complex_numbers(width=32))
elif dtype == complex128:
if finite:
return draw(complex_numbers(allow_nan=False, allow_infinity=False))
return draw(complex_numbers())
else:
raise ValueError(f"Unrecognized dtype {dtype}")

@composite
def array_and_py_scalar(draw, dtypes):
"""Draw a pair: (array, scalar) or (scalar, array)."""
dtype = draw(sampled_from(dtypes))
scalar_var = draw(scalars(just(dtype), finite=True,
**{'min_value': 1/ (2<<5), 'max_value': 2<<5}
))

Note that the [1/64, 64] range exclusively applies to float/complex. The code could use a clarification there as it is not obvious upon first read that integers ignore it.

e.g. pytest -k minimum returns this:

array_api_tests/test_has_names.py::test_has_names[elementwise-minimum] PASSED                                                                                    [ 20%]
array_api_tests/test_operators_and_elementwise_functions.py::test_minimum PASSED                                                                                 [ 40%]
array_api_tests/test_operators_and_elementwise_functions.py::test_binary_with_scalars_real[minimum] SKIPPED (requires ARRAY_API_TESTS_VERSION >= 2024.12)        [ 60%]
array_api_tests/test_signatures.py::test_func_signature[minimum] PASSED                                                                                          [ 80%]
array_api_tests/test_special_cases.py::test_binary[minimum(x1_i is NaN or x2_i is NaN) -> NaN] PASSED   

To my understanding

  • in test_binary_with_scalars_real, when one arg is Array[float64] or Array[float32], the other arg is always in the [1/64, 64] range;
  • test_special_cases::test_binary only tests arrays containing NaN, not scalar NaN;
  • none tests inf or arrays containing inf

related:

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions