Fix test_take to make axis optional when ndim == 1

I didn't explicitly test axis=None because it's not clear to me that should
actually be supported, given that that's the same as axis=0.

Author: asmeurer

array_api_tests/test_indexing_functions.py

@@ -22,15 +22,22 @@ def test_take(x, data):
     # * negative axis
     # * negative indices
     # * different dtypes for indices
-    axis = data.draw(st.integers(0, max(x.ndim - 1, 0)), label="axis")
+
+    # axis is optional but only if x.ndim == 1
+    _axis_st = st.integers(0, max(x.ndim - 1, 0))
+    if x.ndim == 1:
+        kw = data.draw(hh.kwargs(axis=_axis_st))
+    else:
+        kw = {"axis": data.draw(_axis_st)}
+    axis = kw.get("axis", 0)
     _indices = data.draw(
         st.lists(st.integers(0, x.shape[axis] - 1), min_size=1, unique=True),
         label="_indices",
     )
     indices = xp.asarray(_indices, dtype=dh.default_int)
     note(f"{indices=}")
 
-    out = xp.take(x, indices, axis=axis)
+    out = xp.take(x, indices, **kw)
 
     ph.assert_dtype("take", in_dtype=x.dtype, out_dtype=out.dtype)
     ph.assert_shape(