diff options
| author | Charles Harris <charlesr.harris@gmail.com> | 2022-01-13 11:33:17 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-01-13 11:33:17 -0700 |
| commit | 7191d9a4773d77205349ac151f84b72c0ffcf848 (patch) | |
| tree | df7eec8662a2f681ac3bcd20053ba0db864af12d /numpy/array_api/tests/test_set_functions.py | |
| parent | f5e74cb4ee7641f72e36901043a7142dd0ac4dcf (diff) | |
| parent | ccc1091360639de76fab1c1e2a8b31fda81855fb (diff) | |
| download | numpy-7191d9a4773d77205349ac151f84b72c0ffcf848.tar.gz | |
Merge pull request #20791 from honno/xp-inverse-indices
BUG: Return correctly shaped inverse indices in `array_api` set functions
Diffstat (limited to 'numpy/array_api/tests/test_set_functions.py')
| -rw-r--r-- | numpy/array_api/tests/test_set_functions.py | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/numpy/array_api/tests/test_set_functions.py b/numpy/array_api/tests/test_set_functions.py new file mode 100644 index 000000000..b8eb65d43 --- /dev/null +++ b/numpy/array_api/tests/test_set_functions.py @@ -0,0 +1,19 @@ +import pytest +from hypothesis import given +from hypothesis.extra.array_api import make_strategies_namespace + +from numpy import array_api as xp + +xps = make_strategies_namespace(xp) + + +@pytest.mark.parametrize("func", [xp.unique_all, xp.unique_inverse]) +@given(xps.arrays(dtype=xps.scalar_dtypes(), shape=xps.array_shapes())) +def test_inverse_indices_shape(func, x): + """ + Inverse indices share shape of input array + + See https://github.com/numpy/numpy/issues/20638 + """ + out = func(x) + assert out.inverse_indices.shape == x.shape |