Add a test for reductions with where

Author: FrancescAlted

tests/ndarray/test_reductions.py

@@ -65,6 +65,19 @@ def test_reduce_bool(array_fixture, reduce_op):
     np.testing.assert_allclose(res, nres, atol=tol, rtol=tol)
 
 
+def test_reduce_where(array_fixture):
+    a1, a2, a3, a4, na1, na2, na3, na4 = array_fixture
+    # The next works
+    # res = blosc2.where(a1 < a2, a2, 0).sum()
+    # nres = ne_evaluate("sum(where(na1 < na2, na2, 0))")
+    # This does not work yet (it currently hangs)
+    res = blosc2.where(a1 < a2, a2, a1).sum()
+    nres = ne_evaluate("sum(where(na1 < na2, na2, na1))")
+    print("res:", res, nres)
+    tol = 1e-15 if a1.dtype == "float64" else 1e-6
+    np.testing.assert_allclose(res, nres, atol=tol, rtol=tol)
+
+
 @pytest.mark.parametrize(
     "reduce_op", ["sum", "prod", "mean", "std", "var", "min", "max", "any", "all", "argmax", "argmin"]
 )