Add values testing for moveaxis

Author: asmeurer

array_api_tests/test_manipulation_functions.py

@@ -173,7 +173,7 @@ def test_moveaxis(x, data):
 
     ph.assert_dtype("moveaxis", in_dtype=x.dtype, out_dtype=out.dtype)
 
-    # Shape testing
+
     _source = sh.normalize_axis(source, x.ndim)
     _destination = sh.normalize_axis(destination, x.ndim)
 
@@ -188,6 +188,12 @@ def test_moveaxis(x, data):
                            out_shape=out.shape, expected=expected_shape,
                            kw={"source": source, "destination": destination})
 
+    indices = list(sh.ndindex(x.shape))
+    permuted_indices = [tuple(idx[axis] for axis in new_axes) for idx in indices]
+    assert_array_ndindex(
+        "moveaxis", x, x_indices=sh.ndindex(x.shape), out=out, out_indices=permuted_indices
+    )
+
 @pytest.mark.unvectorized
 @given(
     x=hh.arrays(