Name	Size	Mode	Actions
__pycache__/	-	0755	rm
test_array_object.py	17035	0644	editdlrm
test_creation_functions.py	5023	0644	editdlrm
test_data_type_functions.py	1018	0644	editdlrm
test_elementwise_functions.py	3800	0644	editdlrm
test_indexing_functions.py	627	0644	editdlrm
test_manipulation_functions.py	1087	0644	editdlrm
test_set_functions.py	546	0644	editdlrm
test_sorting_functions.py	602	0644	editdlrm
test_validation.py	676	0644	editdlrm
__init__.py	282	0644	editdlrm

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