Commit db7250e6 authored by Markus Holzer's avatar Markus Holzer
Browse files

Fix datahandling test case

parent 243e3f14
Pipeline #35825 passed with stages
in 14 minutes and 56 seconds
......@@ -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
......
......@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment