From 0a1244758bce49bd22320becb9be02c95f2284c7 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Wed, 30 Jun 2021 12:37:43 -0500
Subject: [PATCH] Add test_abs_complex
Co-authored-by: Alex Fikl <alexfikl@gmail.com>
---
test/test_arraycontext.py | 11 +++++++++++
1 file changed, 11 insertions(+)
diff --git a/test/test_arraycontext.py b/test/test_arraycontext.py
index fb21bcb..9cdce37 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:
--
GitLab