From 01b08cb6902ba2946b944b70aaa629a089521537 Mon Sep 17 00:00:00 2001
From: Thomas Gibson <gibsonthomas1120@hotmail.com>
Date: Fri, 21 May 2021 16:14:28 -0500
Subject: [PATCH] Add unit test for concatenating cl arrays
---
test/test_arraycontext.py | 17 +++++++++++++++++
1 file changed, 17 insertions(+)
diff --git a/test/test_arraycontext.py b/test/test_arraycontext.py
index a4fb4bf..ac31d2a 100644
--- a/test/test_arraycontext.py
+++ b/test/test_arraycontext.py
@@ -178,6 +178,23 @@ def test_array_context_np_workalike(actx_factory):
# }}}
+def test_actx_concatenate(actx_factory):
+ actx = actx_factory()
+
+ ndofs = 5000
+ args = [np.random.randn(ndofs) for i in range(10)]
+ ref_result = np.concatenate(args)
+
+ # {{{ test cl.Arrays
+
+ actx_args = [actx.from_numpy(arg) for arg in args]
+ actx_result = actx.to_numpy(actx.np.concatenate(actx_args))
+
+ assert np.allclose(actx_result, ref_result)
+
+ # }}}
+
+
def test_dof_array_arithmetic_same_as_numpy(actx_factory):
actx = actx_factory()
--
GitLab