Test unaffected indices of test_setitem

Author: honno

array_api_tests/test_array_object.py

@@ -88,6 +88,7 @@ def test_setitem(shape, data):
     )
     x = xp.asarray(obj, dtype=dtype)
     note(f"{x=}")
+    # TODO: test setting non-0d arrays
     key = data.draw(xps.indices(shape=shape, max_dims=0), label="key")
     value = data.draw(
         xps.from_dtype(dtype) | xps.arrays(dtype=dtype, shape=()), label="value"
@@ -105,7 +106,18 @@ def test_setitem(shape, data):
         else:
             assert res[key] == value, msg
     else:
-        ph.assert_0d_equals("__setitem__", "value", value, f"x[{key}]", res[key])
+        ph.assert_0d_equals(
+            "__setitem__", "value", value, f"modified x[{key}]", res[key]
+        )
+    _key = key if isinstance(key, tuple) else (key,)
+    assume(all(isinstance(i, int) for i in _key))  # TODO: normalise slices and ellipsis
+    _key = tuple(i if i >= 0 else s + i for i, s in zip(_key, x.shape))
+    unaffected_indices = list(sh.ndindex(res.shape))
+    unaffected_indices.remove(_key)
+    for idx in unaffected_indices:
+        ph.assert_0d_equals(
+            "__setitem__", f"old x[{idx}]", x[idx], f"modified x[{idx}]", res[idx]
+        )
 
 
 # TODO: make mask tests optional
@@ -140,7 +152,6 @@ def test_getitem_mask(shape, data):
         size = int(xp.sum(xp.astype(key, xp.uint8)))
         out_shape = (size,) + x.shape[key.ndim :]
     ph.assert_shape("__getitem__", out.shape, out_shape)
-
     if not any(s == 0 for s in key.shape):
         assume(key.ndim == x.ndim)  # TODO: test key.ndim < x.ndim scenarios
         out_indices = sh.ndindex(out.shape)
@@ -169,7 +180,6 @@ def test_setitem_mask(shape, data):
 
     ph.assert_dtype("__setitem__", x.dtype, res.dtype, repr_name="x.dtype")
     ph.assert_shape("__setitem__", res.shape, x.shape, repr_name="x.dtype")
-
     scalar_type = dh.get_scalar_type(x.dtype)
     for idx in sh.ndindex(x.shape):
         if key[idx]: