Skip to content
Snippets Groups Projects
Commit 053f19d2 authored by Christoph Rettinger's avatar Christoph Rettinger
Browse files

Accomodated API change in pystencils

parent 2de62dcd
No related merge requests found
Pipeline #22082 passed with stage
in 1 minute and 26 seconds
...@@ -34,13 +34,13 @@ class WalberlaLbmpyCodegenTest(unittest.TestCase): ...@@ -34,13 +34,13 @@ class WalberlaLbmpyCodegenTest(unittest.TestCase):
@staticmethod @staticmethod
def test_sparse(): def test_sparse():
from lbmpy.creationfunctions import create_lb_collision_rule from lbmpy.creationfunctions import create_lb_collision_rule
from pystencils import show_code from pystencils import get_code_str
g = ListLbGenerator(create_lb_collision_rule()) g = ListLbGenerator(create_lb_collision_rule())
kernel_code = str(show_code(g.kernel())) kernel_code = get_code_str(g.kernel())
assert 'num_cells' in kernel_code assert 'num_cells' in kernel_code
setter_code = str(show_code(g.setter_ast())) setter_code = get_code_str(g.setter_ast())
assert 'num_cells' in setter_code assert 'num_cells' in setter_code
getter_code = str(show_code(g.getter_ast())) getter_code = get_code_str(g.getter_ast())
assert 'num_cells' in getter_code assert 'num_cells' in getter_code
@staticmethod @staticmethod
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment