From 9bbb8181dcd0717dd61de853f33e11c1ca19d806 Mon Sep 17 00:00:00 2001
From: zy69guqi <richard.angersbach@fau.de>
Date: Fri, 17 Jan 2025 14:56:12 +0100
Subject: [PATCH] Parameterize test_reduction.py for different reduction
operations
---
tests/kernelcreation/test_reduction.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/tests/kernelcreation/test_reduction.py b/tests/kernelcreation/test_reduction.py
index 0532b30f5..c41d250f4 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)
--
GitLab