From e75398f3b75dd158f294c5963341e0ac82686c41 Mon Sep 17 00:00:00 2001
From: Martin Bauer <martin.bauer@fau.de>
Date: Wed, 14 Nov 2018 09:18:59 +0100
Subject: [PATCH] Fixes in codegen files

---
 tests/cuda/codegen/CudaJacobiKernel.py          | 8 ++++----
 tests/lbm/codegen/SrtWithForceFieldModel.gen.py | 7 ++++---
 2 files changed, 8 insertions(+), 7 deletions(-)

diff --git a/tests/cuda/codegen/CudaJacobiKernel.py b/tests/cuda/codegen/CudaJacobiKernel.py
index 225593dcd..14e46d2b6 100644
--- a/tests/cuda/codegen/CudaJacobiKernel.py
+++ b/tests/cuda/codegen/CudaJacobiKernel.py
@@ -3,17 +3,17 @@ from pystencils_walberla.sweep import Sweep
 
 def jacobi2D(sweep):
     src = sweep.field("f1")
-    dst = sweep.temporaryField(src)
+    dst = sweep.temporary_field(src)
 
-    dst[0, 0] @= (src[1, 0] + src[-1, 0] + src[0, 1] + src[0, -1]) / (4 * sweep.constant.h ** 2)
+    dst[0, 0] @= (src[1, 0] + src[-1, 0] + src[0, 1] + src[0, -1]) / (4 * sweep.constant("h") ** 2)
 
 
 def jacobi3D(sweep):
     src = sweep.field("f1")
-    dst = sweep.temporaryField(src)
+    dst = sweep.temporary_field(src)
 
     dst[0, 0, 0] @= (src[1, 0, 0] + src[-1, 0, 0] + src[0, 1, 0] + src[0, -1, 0] + src[0, 0, 1] + src[0, 0, -1]) / \
-                    (6 * sweep.constant.h ** 2)
+                    (6 * sweep.constant("h") ** 2)
 
 
 Sweep.generate('CudaJacobiKernel2D', jacobi2D, dim=2, target='gpu')
diff --git a/tests/lbm/codegen/SrtWithForceFieldModel.gen.py b/tests/lbm/codegen/SrtWithForceFieldModel.gen.py
index f004d3840..72e10eb35 100644
--- a/tests/lbm/codegen/SrtWithForceFieldModel.gen.py
+++ b/tests/lbm/codegen/SrtWithForceFieldModel.gen.py
@@ -1,6 +1,6 @@
 import sympy as sp
 from lbmpy.boundaries import NoSlip, UBB
-from lbmpy_walberla import Field, generate_lattice_model_files, RefinementScaling
+from lbmpy_walberla import generate_lattice_model_files, RefinementScaling
 from lbmpy.creationfunctions import create_lb_method
 from lbmpy_walberla.boundary import create_boundary_class
 from pystencils_walberla.cmake_integration import codegen
@@ -25,11 +25,12 @@ def genBoundary():
     method = create_lb_method(stencil='D3Q19', method='srt')
     return create_boundary_class(boundary, method)
 
+
 def genNoSlip():
     boundary = NoSlip(name='MyNoSlip')
     method = create_lb_method(stencil='D3Q19', method='srt')
     return create_boundary_class(boundary, method)
 
-codegen.register(['MyUBB.h', 'MyUBB.cpp'], genBoundary)
-codegen.register(['MyNoSlip.h', 'MyNoSlip.cpp',], genNoSlip)
 
+codegen.register(['MyUBB.h', 'MyUBB.cpp'], genBoundary)
+codegen.register(['MyNoSlip.h', 'MyNoSlip.cpp'], genNoSlip)
-- 
GitLab