WIP

Author: ev-br

array_api_tests/test_manipulation_functions.py

@@ -156,6 +156,20 @@ def test_expand_dims(x, axis):
         raise
 
 
+@given(
+    x=hh.arrays(dtype=hh.all_dtypes, shape=shared_shapes(max_dims=4)),
+    axes=shared_shapes().flatmap(
+        lambda s: st.lists(
+            st.integers(2*(-len(s)-1), 2*len(s)),
+            min_size=0 if len(s)==0 else 1,
+            max_size=len(s)
+        ).map(tuple)
+    )
+)
+def test_expand_dims_tuples(x, axes):
+    print(x.shape, axes)
+
+
 @pytest.mark.min_version("2023.12")
 @given(x=hh.arrays(dtype=hh.all_dtypes, shape=hh.shapes(min_dims=1)), data=st.data())
 def test_moveaxis(x, data):