test_print_infinity.py

import pytest
pytest.importorskip('graphviz')

from pystencils.session import *
from pystencils.astnodes import Block, Conditional

src, dst = ps.fields("src, dst: double[2D]", layout='c')

true_block = Block([ps.Assignment(dst[0, 0], src[-1, 0])])
false_block = Block([ps.Assignment(dst[0, 0], src[1, 0])])
ur = [true_block, Conditional(dst.center() > 0.0, true_block, false_block)]
    
ast = ps.create_kernel(ur)

ps.to_dot(ast, graph_style={'size': "9.5,12.5"})