add tests for any and all

test/test_arraycontext.py

@@ -120,6 +120,14 @@ class DOFArray:
 
     __array_priority__ = 10
 
+    def __bool__(self):
+        if len(self) == 1 and self.data[0].size == 1:
+            return bool(self.data[0])
+
+        raise ValueError(
+                "The truth value of an array with more than one element is "
+                "ambiguous. Use actx.np.any(x) or actx.np.all(x)")
+
     def __len__(self):
         return len(self.data)
 
@@ -260,6 +268,8 @@ def assert_close_to_numpy_in_containers(actx, op, args):
             ("where", 3, np.float64),
             ("min", 1, np.float64),
             ("max", 1, np.float64),
+            ("any", 1, np.float64),
+            ("all", 1, np.float64),
 
             # float + complex
             ("sin", 1, np.float64),
@@ -519,6 +529,24 @@ def test_reductions_same_as_numpy(actx_factory, op):
 
     assert np.allclose(np_red, actx_red)
 
+
+@pytest.mark.parametrize("sym_name", ["any", "all"])
+def test_any_all_same_as_numpy(actx_factory, sym_name):
+    actx = actx_factory()
+    if not hasattr(actx.np, sym_name):
+        pytest.skip(f"'{sym_name}' not implemented on '{type(actx).__name__}'")
+
+    rng = np.random.default_rng()
+    ary_any = rng.integers(0, 2, 512)
+    ary_all = np.ones(512)
+
+    assert_close_to_numpy_in_containers(actx,
+                lambda _np, *_args: getattr(_np, sym_name)(*_args), [ary_any])
+    assert_close_to_numpy_in_containers(actx,
+                lambda _np, *_args: getattr(_np, sym_name)(*_args), [ary_all])
+    assert_close_to_numpy_in_containers(actx,
+                lambda _np, *_args: getattr(_np, sym_name)(*_args), [1 - ary_all])
+
 # }}}