test

conftest.py

@@ -82,10 +82,6 @@ try:
 except ImportError:
     collect_ignore += [os.path.join(SCRIPT_FOLDER, "pystencils/datahandling/vtk.py")]
 
-# TODO: Remove if Ubuntu 18.04 is no longer supported
-if pytest_version < 50403:
-    collect_ignore += [os.path.join(SCRIPT_FOLDER, "pystencils_tests/test_jupyter_extensions.ipynb")]
-
 collect_ignore += [os.path.join(SCRIPT_FOLDER, 'setup.py')]
 
 for root, sub_dirs, files in os.walk('.'):

pystencils_tests/test_vectorization_specific.py

@@ -63,7 +63,7 @@ def test_vectorized_abs(instruction_set, dtype):
 def test_strided(instruction_set, dtype):
     npdtype = np.float64 if dtype == 'double' else np.float32
     type_string = "float64" if dtype == 'double' else "float32"
-    
+
     f, g = ps.fields(f"f, g : {type_string}[2D]")
     update_rule = [ps.Assignment(g[0, 0], f[0, 0] + f[-1, 0] + f[1, 0] + f[0, 1] + f[0, -1] + 42.0)]
     if 'storeS' not in get_vector_instruction_set(dtype, instruction_set) and instruction_set not in ['avx512', 'rvv'] and not instruction_set.startswith('sve'):
@@ -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,7 +92,7 @@ def test_strided(instruction_set, dtype):
     func(g=dst, f=arr)
     ref_func(g=ref, f=arr)
 
-    np.testing.assert_almost_equal(dst, ref, 13 if dtype == 'double' else 5)
+    np.testing.assert_almost_equal(dst[1:-1, 1:-1], ref[1:-1, 1:-1], 13 if dtype == 'double' else 5)
 
 
 @pytest.mark.parametrize('dtype', ('float', 'double'))