migrate test suite

tests/test_fvm.py → tests/_todo/test_fvm.py

@@ -4,8 +4,8 @@ import numpy as np
 import pytest
 from itertools import product
 from pystencils.rng import random_symbol
-from pystencils.sympyextensions.astnodes import SympyAssignment
-from pystencils.node_collection import NodeCollection
+# from pystencils.sympyextensions.astnodes import SympyAssignment
+# from pystencils.node_collection import NodeCollection
 
 
 def advection_diffusion(dim: int):

tests/test_indexed_kernels.py → tests/_todo/test_indexed_kernels.py

@@ -8,39 +8,6 @@ from pystencils.transformations import filtered_tree_iteration
 from pystencils.typing import BasicType, FieldPointerSymbol, PointerType, TypedSymbol
 
 
-@pytest.mark.parametrize('target', [ps.Target.CPU, ps.Target.GPU])
-def test_indexed_kernel(target):
-    if target == Target.GPU:
-        pytest.importorskip("cupy")
-        import cupy as cp
-
-    arr = np.zeros((3, 4))
-    dtype = np.dtype([('x', int), ('y', int), ('value', arr.dtype)])
-    index_arr = np.zeros((3,), dtype=dtype)
-    index_arr[0] = (0, 2, 3.0)
-    index_arr[1] = (1, 3, 42.0)
-    index_arr[2] = (2, 1, 5.0)
-
-    indexed_field = Field.create_from_numpy_array('index', index_arr)
-    normal_field = Field.create_from_numpy_array('f', arr)
-    update_rule = Assignment(normal_field[0, 0], indexed_field('value'))
-
-    config = CreateKernelConfig(target=target, index_fields=[indexed_field])
-    ast = create_kernel([update_rule], config=config)
-    kernel = ast.compile()
-
-    if target == Target.CPU:
-        kernel(f=arr, index=index_arr)
-    else:
-        gpu_arr = cp.asarray(arr)
-        gpu_index_arr = cp.ndarray(index_arr.shape, dtype=index_arr.dtype)
-        gpu_index_arr.set(index_arr)
-        kernel(f=gpu_arr, index=gpu_index_arr)
-        arr = gpu_arr.get()
-    for i in range(index_arr.shape[0]):
-        np.testing.assert_allclose(arr[index_arr[i]['x'], index_arr[i]['y']], index_arr[i]['value'], atol=1e-13)
-
-
 @pytest.mark.parametrize('index_size', ("fixed", "variable"))
 @pytest.mark.parametrize('array_size', ("3D", "2D", "10, 12", "13, 17, 19"))
 @pytest.mark.parametrize('target', (ps.Target.CPU, ps.Target.GPU))