diff --git a/tests/kernelcreation/test_reduction.py b/tests/kernelcreation/test_reduction.py index 0532b30f552e4800fb5fb3899d50ad539f46c0d4..c41d250f4d75e236dad9a6d6bd476045c2847ad0 100644 --- a/tests/kernelcreation/test_reduction.py +++ b/tests/kernelcreation/test_reduction.py @@ -3,17 +3,18 @@ import numpy as np import sympy as sp import pystencils as ps -from pystencils import AddReducedAssignment +from pystencils.sympyextensions import reduced_assign @pytest.mark.parametrize('dtype', ["float64"]) -def test_reduction(dtype): +@pytest.mark.parametrize("op", ["+", "-", "*", "min", "max"]) +def test_reduction(dtype, op): x = ps.fields(f'x: {dtype}[1d]') w = sp.Symbol("w") # kernel with reduction assignment - reduction_assignment = AddReducedAssignment(w, x.center()) + reduction_assignment = reduced_assign(w, op, x.center()) config = ps.CreateKernelConfig(cpu_openmp=True)