From c2a52c2828643c80bc3e48e7c0b66eac4b8bfe4e Mon Sep 17 00:00:00 2001
From: Martin Bauer <martin.bauer@fau.de>
Date: Mon, 29 Oct 2018 15:58:48 +0100
Subject: [PATCH] Increased test coverage
---
__init__.py | 5 ++++-
generate_packinfo.py | 18 ------------------
2 files changed, 4 insertions(+), 19 deletions(-)
diff --git a/__init__.py b/__init__.py
index d825522..dba04d0 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1 +1,4 @@
-from .sweep import Sweep
\ No newline at end of file
+from .sweep import Sweep
+from .generate_packinfo import generate_pack_info_from_kernel, generate_pack_info_for_field
+
+__all__ = ['Sweep', 'generate_pack_info_from_kernel', 'generate_pack_info_for_field']
diff --git a/generate_packinfo.py b/generate_packinfo.py
index 7cc6a5e..4d9bbd0 100644
--- a/generate_packinfo.py
+++ b/generate_packinfo.py
@@ -106,21 +106,3 @@ def generate_pack_info(class_name: str,
header = env.get_template("GpuPackInfo.tmpl.h").render(**context)
source = env.get_template("GpuPackInfo.tmpl.cpp").render(**context)
return header, source
-
-
-if __name__ == '__main__':
- from lbmpy.creationfunctions import create_lb_update_rule
- ur = create_lb_update_rule(stencil='D2Q9')
- header, source = generate_pack_info_from_kernel('D2Q9PackInfo', ur.main_assignments, target='gpu')
- print(header, file=open('/local/bauer/code/walberla/tests/cuda/D2Q9PackInfo.h', 'w'))
- print(source, file=open('/local/bauer/code/walberla/tests/cuda/D2Q9PackInfo.cu', 'w'))
- print("Done")
-
-
-if __name__ == '__main__2':
- f = Field.create_generic('f', spatial_dimensions=3, index_dimensions=0, layout='fzyx')
-
- header, source = generate_pack_info_for_field('GenGpuPackInfo', f, target='gpu')
- print(header, file=open('/local/bauer/code/walberla/tests/cuda/GenGpuPackInfo.h', 'w'))
- print(source, file=open('/local/bauer/code/walberla/tests/cuda/GenGpuPackInfo.cu', 'w'))
- print("Done")
--
GitLab