From 7e32f00c90fb5120d35a06c5076c03946ab7cc40 Mon Sep 17 00:00:00 2001
From: Markus Holzer <markus.holzer@fau.de>
Date: Fri, 23 Jun 2023 16:17:19 +0200
Subject: [PATCH] Add more tests
---
pystencils_tests/test_datahandling.py | 15 +++++++++++++++
1 file changed, 15 insertions(+)
diff --git a/pystencils_tests/test_datahandling.py b/pystencils_tests/test_datahandling.py
index 5c8c8c23..7f73a865 100644
--- a/pystencils_tests/test_datahandling.py
+++ b/pystencils_tests/test_datahandling.py
@@ -387,3 +387,18 @@ def test_array_handler():
random_array = array_handler.randn(size)
+ cpu_array = np.empty((20, 40), dtype=np.float64)
+ gpu_array = array_handler.to_gpu(cpu_array)
+
+ assert cpu_array.base is None
+ assert gpu_array.base is None
+ assert gpu_array.strides == cpu_array.strides
+
+ cpu_array2 = np.empty((20, 40), dtype=np.float64)
+ cpu_array2 = cpu_array2.swapaxes(0, 1)
+ gpu_array2 = array_handler.to_gpu(cpu_array2)
+
+ assert cpu_array2.base is not None
+ assert gpu_array2.base is not None
+ assert gpu_array2.strides == cpu_array2.strides
+
--
GitLab