def test_stencils_roundtrip(stencil_name):
stencil_def = stencil_registry[stencil_name]
externals = externals_registry[stencil_name]
oir = stencil_def_to_oir(stencil_def, externals)
sdfg = OirSDFGBuilder().visit(oir)
sdfg_pre = deepcopy(sdfg)
oir = convert(sdfg)
sdfg_post = OirSDFGBuilder().visit(oir)
assert_sdfg_equal(sdfg_pre, sdfg_post)
def __eq__(self, other):
try:
assert isinstance(other, VerticalLoopLibraryNode)
assert self.loop_order == other.loop_order
assert self.caches == other.caches
assert len(self.sections) == len(other.sections)
for (interval1,
he_sdfg1), (interval2,
he_sdfg2) in zip(self.sections, other.sections):
assert interval1 == interval2
assert_sdfg_equal(he_sdfg1, he_sdfg2)
except AssertionError:
return False
return True