diff --git a/test/test_arraycontext.py b/test/test_arraycontext.py index fb21bcb5aa8373dc4dfba42a45faa50abab75629..9cdce371f0ddbf7b97c44406446fa759d5612a90 100644 --- a/test/test_arraycontext.py +++ b/test/test_arraycontext.py @@ -784,6 +784,17 @@ def test_container_equality(actx_factory): assert isinstance(bcast_dc_of_dofs == bcast_dc_of_dofs_2, MyContainerDOFBcast) +def test_abs_complex(actx_factory): + actx = actx_factory() + a = np.random.randn(2000) + 1j * np.random.randn(2000) + + abs_a_ref = np.abs(a) + abs_a = actx.np.abs(actx.from_numpy(a)) + + assert abs_a.dtype == abs_a_ref.dtype + np.testing.assert_allclose(actx.to_numpy(abs_a), abs_a_ref) + + if __name__ == "__main__": import sys if len(sys.argv) > 1: