From ccc1091360639de76fab1c1e2a8b31fda81855fb Mon Sep 17 00:00:00 2001
From: Matthew <quitesimplymatt@gmail.com>
Date: Thu, 13 Jan 2022 10:03:49 +0000
Subject: [PATCH] Regression test for inverse indices in `array_api` set
 functions

---
 numpy/array_api/tests/test_set_functions.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)
 create mode 100644 numpy/array_api/tests/test_set_functions.py

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 000000000000..b8eb65d430dc
--- /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