From db7250e65aaa4b49f1021ec734137ed6ac4463ee Mon Sep 17 00:00:00 2001
From: Markus Holzer <markus.holzer@fau.de>
Date: Mon, 22 Nov 2021 14:27:53 +0100
Subject: [PATCH] Fix datahandling test case
---
pystencils/rng.py | 1 -
pystencils_tests/test_datahandling.py | 8 +++-----
2 files changed, 3 insertions(+), 6 deletions(-)
diff --git a/pystencils/rng.py b/pystencils/rng.py
index 863867bf..7c4f894f 100644
--- a/pystencils/rng.py
+++ b/pystencils/rng.py
@@ -5,7 +5,6 @@ import sympy as sp
from pystencils.data_types import TypedSymbol, cast_func
from pystencils.astnodes import LoopOverCoordinate
from pystencils.backends.cbackend import CustomCodeNode
-from pystencils.enums import Backend
from pystencils.sympyextensions import fast_subs
diff --git a/pystencils_tests/test_datahandling.py b/pystencils_tests/test_datahandling.py
index f160f5f5..be695d07 100644
--- a/pystencils_tests/test_datahandling.py
+++ b/pystencils_tests/test_datahandling.py
@@ -357,12 +357,10 @@ def test_load_data():
assert np.all(dh.cpu_arrays['dst2']) == 0
-@pytest.mark.parametrize('target', (Target.GPU))
-def test_array_handler(target):
+def test_array_handler():
size = (2, 2)
- if target == Target.GPU:
- pytest.importorskip('pycuda')
- array_handler = PyCudaArrayHandler()
+ pytest.importorskip('pycuda')
+ array_handler = PyCudaArrayHandler()
zero_array = array_handler.zeros(size)
cpu_array = np.empty(size)
--
GitLab