Skip to content
Snippets Groups Projects
Commit db7250e6 authored by Markus Holzer's avatar Markus Holzer
Browse files

Fix datahandling test case

parent 243e3f14
No related branches found
No related tags found
No related merge requests found
Pipeline #35825 passed
...@@ -5,7 +5,6 @@ import sympy as sp ...@@ -5,7 +5,6 @@ import sympy as sp
from pystencils.data_types import TypedSymbol, cast_func from pystencils.data_types import TypedSymbol, cast_func
from pystencils.astnodes import LoopOverCoordinate from pystencils.astnodes import LoopOverCoordinate
from pystencils.backends.cbackend import CustomCodeNode from pystencils.backends.cbackend import CustomCodeNode
from pystencils.enums import Backend
from pystencils.sympyextensions import fast_subs from pystencils.sympyextensions import fast_subs
......
...@@ -357,12 +357,10 @@ def test_load_data(): ...@@ -357,12 +357,10 @@ def test_load_data():
assert np.all(dh.cpu_arrays['dst2']) == 0 assert np.all(dh.cpu_arrays['dst2']) == 0
@pytest.mark.parametrize('target', (Target.GPU)) def test_array_handler():
def test_array_handler(target):
size = (2, 2) size = (2, 2)
if target == Target.GPU: pytest.importorskip('pycuda')
pytest.importorskip('pycuda') array_handler = PyCudaArrayHandler()
array_handler = PyCudaArrayHandler()
zero_array = array_handler.zeros(size) zero_array = array_handler.zeros(size)
cpu_array = np.empty(size) cpu_array = np.empty(size)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment