def test_vertical_index_fields(): validate(pyast_to_sir(callable_to_pyast(simple_example))) # FIXME: add validation again pyast_to_sir(callable_to_pyast(various_expression)) pyast_to_sir(callable_to_pyast(index_fields_with_offsets)) pyast_to_sir(callable_to_pyast(various_dimensions_mix)) pyast_to_sir(callable_to_pyast(sparse_index_fields))
def test_math(): validate(pyast_to_sir(callable_to_pyast(math_stencil)))
def test_sparse_fill(): validate(pyast_to_sir(callable_to_pyast(sparse_order_2_fill))) validate(pyast_to_sir(callable_to_pyast(longer_fills))) validate(pyast_to_sir(callable_to_pyast(fill_with_reduction))) validate(pyast_to_sir(callable_to_pyast(ambiguous_fill))) validate(pyast_to_sir(callable_to_pyast(fill_with_center)))
def test_field(): validate(pyast_to_sir(callable_to_pyast(temp_field))) validate(pyast_to_sir(callable_to_pyast(temp_field_demoted))) validate(pyast_to_sir(callable_to_pyast(hv_field))) validate(pyast_to_sir(callable_to_pyast(h_offsets))) validate(pyast_to_sir(callable_to_pyast(v_offsets))) validate(pyast_to_sir(callable_to_pyast(hv_offsets))) validate( pyast_to_sir(callable_to_pyast(redundant_vertical_index_in_2d_field)))
def test_examples(): validate(pyast_to_sir(callable_to_pyast(laplacian_fd))) validate(pyast_to_sir(callable_to_pyast(laplacian_fvm))) validate(pyast_to_sir(callable_to_pyast(interpolation_sph)))
def test_reduce(): validate(pyast_to_sir(callable_to_pyast(various_reductions))) validate(pyast_to_sir(callable_to_pyast(kw_args))) validate(pyast_to_sir(callable_to_pyast(reductions_with_center)))
def test_simple(): validate(pyast_to_sir(callable_to_pyast(control_flow))) validate(pyast_to_sir(callable_to_pyast(compound_assignment))) validate(pyast_to_sir(callable_to_pyast(power_operator))) validate(pyast_to_sir(callable_to_pyast(vertical_iteration_variable)))
def test_copy(): validate(pyast_to_sir(callable_to_pyast(copy_only_vertex))) validate(pyast_to_sir(callable_to_pyast(copy_only_edge))) validate(pyast_to_sir(callable_to_pyast(copy_only_cell))) validate(pyast_to_sir(callable_to_pyast(copy_all_separate))) validate(pyast_to_sir(callable_to_pyast(copy_all_together)))
def test_examples(): validate(pyast_to_sir(callable_to_pyast(laplacian_fd))) validate(pyast_to_sir(callable_to_pyast(laplacian_fvm)))