diff --git a/pystencils_tests/test_vectorization_specific.py b/pystencils_tests/test_vectorization_specific.py index 088ef913d4a0055ebc9808c574df85deb6eb8a72..7929266d9ca47b7e8987b9431882d7f946c7bd9d 100644 --- a/pystencils_tests/test_vectorization_specific.py +++ b/pystencils_tests/test_vectorization_specific.py @@ -79,7 +79,7 @@ def test_strided(instruction_set, dtype): ast = ps.create_kernel(update_rule, config=config) assert len(warn) == 0 - # ps.show_code(ast) + ps.show_code(ast) func = ast.compile() ref_config = pystencils.config.CreateKernelConfig(default_number_float=type_string) ref_func = ps.create_kernel(update_rule, config=ref_config).compile() @@ -92,6 +92,9 @@ def test_strided(instruction_set, dtype): func(g=dst, f=arr) ref_func(g=ref, f=arr) + print("dst: ", dst) + print("np array: ", arr) + np.testing.assert_almost_equal(dst[1:-1, 1:-1], ref[1:-1, 1:-1], 13 if dtype == 'double' else 5)