adds test_leaf_array_type_broadcasting

Author: kaushikcfd

test/test_arraycontext.py

@@ -765,6 +765,32 @@ def test_norm_ord_none(actx_factory, ndim):
     np.testing.assert_allclose(actx.to_numpy(norm_a), norm_a_ref)
 
 
+@with_container_arithmetic(
+    bcast_obj_array=True,
+    bcast_numpy_array=True,
+    rel_comparison=True,
+    _cls_has_array_context_attr=True)
+@dataclass_array_container
+@dataclass(frozen=True)
+class Foo:
+    u: DOFArray
+
+    @property
+    def array_context(self):
+        return self.u.array_context
+
+
+def test_leaf_array_type_broadcasting(actx_factory):
+    actx = actx_factory()
+
+    foo = Foo(DOFArray(actx, (actx.zeros(3, dtype=np.float64) + 41, )))
+    bar = foo + 4
+    baz = foo + actx.from_numpy(4*np.ones((3, )))
+
+    np.testing.assert_allclose(bar.u[0].get(),
+                               baz.u[0].get())
+
+
 if __name__ == "__main__":
     import sys
     if len(sys.argv) > 1: