From a9598b199b2a944b50266eed56b0285129e3066c Mon Sep 17 00:00:00 2001
From: Frederik Hennig <frederik.hennig@fau.de>
Date: Thu, 16 Nov 2023 12:48:55 +0900
Subject: [PATCH] cleanup in cmake integration test
---
tests/cmake_integration/kernels.py | 12 ++++++------
tests/cmake_integration/more_kernels.py | 12 ++++++------
2 files changed, 12 insertions(+), 12 deletions(-)
diff --git a/tests/cmake_integration/kernels.py b/tests/cmake_integration/kernels.py
index d780b8f..901bdb0 100644
--- a/tests/cmake_integration/kernels.py
+++ b/tests/cmake_integration/kernels.py
@@ -1,20 +1,20 @@
+# type: ignore
+
import sympy as sp
-import numpy as np
-from pystencils.session import *
+from pystencils import fields, kernel
from pystencilssfg import SourceFileGenerator
-from pystencilssfg.source_concepts.cpp import std_mdspan
with SourceFileGenerator() as sfg:
- src, dst = ps.fields("src, dst(1) : double[2D]")
+ src, dst = fields("src, dst(1) : double[2D]")
h = sp.Symbol('h')
- @ps.kernel
+ @kernel
def poisson_jacobi():
- dst[0,0] @= (src[1, 0] + src[-1, 0] + src[0, 1] + src[0, -1]) / 4
+ dst[0, 0] @= (src[1, 0] + src[-1, 0] + src[0, 1] + src[0, -1]) / 4
poisson_kernel = sfg.kernels.create(poisson_jacobi)
diff --git a/tests/cmake_integration/more_kernels.py b/tests/cmake_integration/more_kernels.py
index 2d97a94..75b798f 100644
--- a/tests/cmake_integration/more_kernels.py
+++ b/tests/cmake_integration/more_kernels.py
@@ -1,20 +1,20 @@
+# type: ignore
+
import sympy as sp
-import numpy as np
-from pystencils.session import *
+from pystencils import fields, kernel, Field
from pystencilssfg import SourceFileGenerator
-from pystencilssfg.source_concepts.cpp import std_mdspan
with SourceFileGenerator() as sfg:
- src = ps.fields("src: double[2D]")
+ src: Field = fields("src: double[2D]")
h = sp.Symbol('h')
- @ps.kernel
+ @kernel
def poisson_gs():
- src[0,0] @= (src[1, 0] + src[-1, 0] + src[0, 1] + src[0, -1]) / 4
+ src[0, 0] @= (src[1, 0] + src[-1, 0] + src[0, 1] + src[0, -1]) / 4
poisson_kernel = sfg.kernels.create(poisson_gs)
--
GitLab