ENH: test floor_div, remainder with scalars

Author: ev-br

array_api_tests/test_operators_and_elementwise_functions.py

@@ -1791,6 +1791,10 @@ def _check_binary_with_scalars(func_data, x1x2):
     )
 
 
+def _filter_zero(x):
+    return x != 0 if dh.is_scalar(x) else (not xp.any(x == 0))
+
+
 @pytest.mark.min_version("2024.12")
 @pytest.mark.parametrize('func_data',
     # xp_func, name, refimpl, kwargs, expected_dtype
@@ -1812,11 +1816,19 @@ def _check_binary_with_scalars(func_data, x1x2):
         (xp.less_equal, "les_equal", operator.le, {}, xp.bool),
         (xp.greater, "greater", operator.gt, {}, xp.bool),
         (xp.greater_equal, "greater_equal", operator.ge, {}, xp.bool),
+        (xp.remainder, "remainder", operator.mod, {}, None),
+        (xp.floor_divide, "floor_divide", operator.floordiv, {}, None),
     ],
     ids=lambda func_data: func_data[1]  # use names for test IDs
 )
 @given(x1x2=hh.array_and_py_scalar(dh.real_float_dtypes))
 def test_binary_with_scalars_real(func_data, x1x2):
+
+    if func_data[1] == "remainder":
+        assume(_filter_zero(x1x2[1]))
+    if func_data[1] == "floor_divide":
+        assume(_filter_zero(x1x2[0]) and _filter_zero(x1x2[1]))
+
     _check_binary_with_scalars(func_data, x1x2)