allow a sequence of dtypes in the scalars(...) strategy

use it to simplify the searchorted_with_scalars test

Author: ev-br

array_api_tests/hypothesis_helpers.py

@@ -454,9 +454,12 @@ def scalars(draw, dtypes, finite=False, **kwds):
     """
     Strategy to generate a scalar that matches a dtype strategy
 
-    dtypes should be one of the shared_* dtypes strategies.
+    dtypes should be one of the shared_* dtypes strategies or a sequence of dtypes.
     """
-    dtype = draw(dtypes)
+    if isinstance(dtypes, Sequence):
+        dtype = draw(sampled_from(dtypes))
+    else:
+        dtype = draw(dtypes)
     mM = kwds.pop('mM', None)
     if dh.is_int_dtype(dtype):
         if mM is None:

array_api_tests/test_searching_functions.py

@@ -316,7 +316,7 @@ def test_searchsorted_with_scalars(data):
     kw = data.draw(hh.kwargs(side=st.sampled_from(["left", "right"])))
 
     # 2. draw x2, a real-valued scalar (IOW, an int or a float)
-    x2 = data.draw(hh.scalars(st.sampled_from([xp.int32, xp.float64]), finite=True))
+    x2 = data.draw(hh.scalars([xp.int32, xp.float64], finite=True))
 
     # 3. testing: similar to test_searchsorted, modulo `out.shape == ()`
     repro_snippet = ph.format_snippet(