Hypothesis strategies for Array API libraries

Note: hypothesis-array-api uses private APIs from Hypothesis and so should be considered unstable.

Install

You can get the strategies from PyPI.

pip install hypothesis-array-api

To install from source, get Poetry and then poetry install inside the repository. Using poetry shell is a good idea for development, where you can use pytest to run the full test suite (note there a lot of expected warnings I need to declutter.)

Quickstart

from numpy import array_api as xp

from hypothesis import given
from hypothesis_array import get_strategies_namespace

xps = get_strategies_namespace(xp)

@given(xps.arrays(dtype=xps.scalar_strategies(), shape=xps.array_shapes()))
def your_test(array):
    ...

Contributors

@honno created these strategies with input from @mattip, @asmeurer, @rgommers and other great folk from @Quansight-Labs.

Inspiration was taken from the NumPy strategies that Hypothesis ships with at hypothesis.extra.numpy. Thanks to the Hypothesis contributors who helped shape it, including: @Zac-HD, @rsokl, @DRMacIver, @takluyver, @rdturnermtl, @kprzybyla, @sobolevn, @kir0ul, @lmount, @jdufresne, @gsnsw-felixs and @alexwlchan.