ENH: one more example of the failure reporting

$ ARRAY_API_TESTS_MODULE=array_api_compat.torch pytest array_api_tests/test_array_object.py::test_getitem

Author: ev-br

array_api_tests/test_array_object.py

@@ -85,24 +85,29 @@ def test_getitem(shape, dtype, data):
     note(f"{x=}")
     key = data.draw(xps.indices(shape=shape, allow_newaxis=True), label="key")
 
-    out = x[key]
-
-    ph.assert_dtype("__getitem__", in_dtype=x.dtype, out_dtype=out.dtype)
-    _key = normalize_key(key, shape)
-    axes_indices, expected_shape = get_indexed_axes_and_out_shape(_key, shape)
-    ph.assert_shape("__getitem__", out_shape=out.shape, expected=expected_shape)
-    out_zero_sided = any(side == 0 for side in expected_shape)
-    if not zero_sided and not out_zero_sided:
-        out_obj = []
-        for idx in product(*axes_indices):
-            val = obj
-            for i in idx:
-                val = val[i]
-            out_obj.append(val)
-        out_obj = sh.reshape(out_obj, expected_shape)
-        expected = xp.asarray(out_obj, dtype=dtype)
-        ph.assert_array_elements("__getitem__", out=out, expected=expected)
-
+    repro_snippet = ph.format_snippet(f"{x!r}[{key!r}]")
+
+    try:
+        out = x[key]
+
+        ph.assert_dtype("__getitem__", in_dtype=x.dtype, out_dtype=out.dtype)
+        _key = normalize_key(key, shape)
+        axes_indices, expected_shape = get_indexed_axes_and_out_shape(_key, shape)
+        ph.assert_shape("__getitem__", out_shape=out.shape, expected=expected_shape)
+        out_zero_sided = any(side == 0 for side in expected_shape)
+        if not zero_sided and not out_zero_sided:
+            out_obj = []
+            for idx in product(*axes_indices):
+                val = obj
+                for i in idx:
+                    val = val[i]
+                out_obj.append(val)
+            out_obj = sh.reshape(out_obj, expected_shape)
+            expected = xp.asarray(out_obj, dtype=dtype)
+            ph.assert_array_elements("__getitem__", out=out, expected=expected)
+    except Exception as exc:
+        exc.add_note(repro_snippet)
+        raise
 
 @pytest.mark.unvectorized
 @given(