Smoke searching functions

Author: honno

array_api_tests/test_searching_functions.py

@@ -0,0 +1,41 @@
+from hypothesis import given
+from hypothesis import strategies as st
+
+from . import _array_module as xp
+from . import hypothesis_helpers as hh
+from . import xps
+
+
+# TODO: generate kwargs
+@given(xps.arrays(dtype=xps.scalar_dtypes(), shape=hh.shapes(min_side=1)))
+def test_argmin(x):
+    xp.argmin(x)
+    # TODO
+
+
+# TODO: generate kwargs
+@given(xps.arrays(dtype=xps.scalar_dtypes(), shape=hh.shapes(min_side=1)))
+def test_argmax(x):
+    xp.argmax(x)
+    # TODO
+
+
+# TODO: generate kwargs, skip if opted out
+@given(xps.arrays(dtype=xps.scalar_dtypes(), shape=hh.shapes(min_side=1)))
+def test_nonzero(x):
+    xp.nonzero(x)
+    # TODO
+
+
+# TODO: skip if opted out
+@given(
+    shapes=hh.mutually_broadcastable_shapes(3),
+    dtypes=hh.mutually_promotable_dtypes(),
+    data=st.data(),
+)
+def test_where(shapes, dtypes, data):
+    cond = data.draw(xps.arrays(dtype=xp.bool, shape=shapes[0]), label="condition")
+    x1 = data.draw(xps.arrays(dtype=dtypes[0], shape=shapes[1]), label="x1")
+    x2 = data.draw(xps.arrays(dtype=dtypes[1], shape=shapes[2]), label="x2")
+    xp.where(cond, x1, x2)
+    # TODO

array_api_tests/test_statistical_functions.py

@@ -5,49 +5,49 @@
 from . import xps
 
 
-# TODO generate kwargs
+# TODO: generate kwargs
 @given(xps.arrays(dtype=xps.numeric_dtypes(), shape=hh.shapes(min_side=1)))
 def test_min(x):
     xp.min(x)
     # TODO
 
 
-# TODO generate kwargs
+# TODO: generate kwargs
 @given(xps.arrays(dtype=xps.numeric_dtypes(), shape=hh.shapes(min_side=1)))
 def test_max(x):
     xp.max(x)
     # TODO
 
 
-# TODO generate kwargs
+# TODO: generate kwargs
 @given(xps.arrays(dtype=xps.floating_dtypes(), shape=hh.shapes(min_side=1)))
 def test_mean(x):
     xp.mean(x)
     # TODO
 
 
-# TODO generate kwargs
+# TODO: generate kwargs
 @given(xps.arrays(dtype=xps.numeric_dtypes(), shape=hh.shapes(min_side=1)))
 def test_prod(x):
     xp.prod(x)
     # TODO
 
 
-# TODO generate kwargs
+# TODO: generate kwargs
 @given(xps.arrays(dtype=xps.floating_dtypes(), shape=hh.shapes(min_side=1)))
 def test_std(x):
     xp.std(x)
     # TODO
 
 
-# TODO generate kwargs
+# TODO: generate kwargs
 @given(xps.arrays(dtype=xps.numeric_dtypes(), shape=hh.shapes(min_side=1)))
 def test_sum(x):
     xp.sum(x)
     # TODO
 
 
-# TODO generate kwargs
+# TODO: generate kwargs
 @given(xps.arrays(dtype=xps.floating_dtypes(), shape=hh.shapes(min_side=1)))
 def test_var(x):
     xp.var(x)

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, where, tensordor, vecdot
+# result_type, meshgrid, tensordor, vecdot
 
 
 @given(hh.mutually_promotable_dtypes(None))