Move manipulation tests to their own file

Author: honno

array_api_tests/test_manipulation_functions.py

@@ -0,0 +1,38 @@
+from hypothesis import given
+from hypothesis import strategies as st
+
+from . import _array_module as xp
+from . import dtype_helpers as dh
+from . import hypothesis_helpers as hh
+from . import pytest_helpers as ph
+from . import xps
+
+
+@given(
+    shape=hh.shapes(min_dims=1),
+    dtypes=hh.mutually_promotable_dtypes(None, dtypes=dh.numeric_dtypes),
+    data=st.data(),
+)
+def test_concat(shape, dtypes, data):
+    arrays = []
+    for i, dtype in enumerate(dtypes, 1):
+        x = data.draw(xps.arrays(dtype=dtype, shape=shape), label=f"x{i}")
+        arrays.append(x)
+    out = xp.concat(arrays)
+    ph.assert_dtype("concat", dtypes, out.dtype)
+    # TODO
+
+
+@given(
+    shape=hh.shapes(),
+    dtypes=hh.mutually_promotable_dtypes(None),
+    data=st.data(),
+)
+def test_stack(shape, dtypes, data):
+    arrays = []
+    for i, dtype in enumerate(dtypes, 1):
+        x = data.draw(xps.arrays(dtype=dtype, shape=shape), label=f"x{i}")
+        arrays.append(x)
+    out = xp.stack(arrays)
+    ph.assert_dtype("stack", dtypes, out.dtype)
+    # TODO

array_api_tests/test_type_promotion.py

@@ -19,7 +19,7 @@
 
 
 # TODO: move tests not covering elementwise funcs/ops into standalone tests
-# result_type, meshgrid, concat, stack, where, tensordor, vecdot
+# result_type, meshgrid, where, tensordor, vecdot
 
 
 @given(hh.mutually_promotable_dtypes(None))
@@ -51,34 +51,6 @@ def test_meshgrid(dtypes, data):
         ph.assert_dtype("meshgrid", dtypes, x.dtype, repr_name=f"out[{i}].dtype")
 
 
-@given(
-    shape=hh.shapes(min_dims=1),
-    dtypes=hh.mutually_promotable_dtypes(None, dtypes=dh.numeric_dtypes),
-    data=st.data(),
-)
-def test_concat(shape, dtypes, data):
-    arrays = []
-    for i, dtype in enumerate(dtypes, 1):
-        x = data.draw(xps.arrays(dtype=dtype, shape=shape), label=f"x{i}")
-        arrays.append(x)
-    out = xp.concat(arrays)
-    ph.assert_dtype("concat", dtypes, out.dtype)
-
-
-@given(
-    shape=hh.shapes(),
-    dtypes=hh.mutually_promotable_dtypes(None),
-    data=st.data(),
-)
-def test_stack(shape, dtypes, data):
-    arrays = []
-    for i, dtype in enumerate(dtypes, 1):
-        x = data.draw(xps.arrays(dtype=dtype, shape=shape), label=f"x{i}")
-        arrays.append(x)
-    out = xp.stack(arrays)
-    ph.assert_dtype("stack", dtypes, out.dtype)
-
-
 bitwise_shift_funcs = [
     "bitwise_left_shift",
     "bitwise_right_shift",