diff --git a/pystencils_tests/test_interpolation.py b/pystencils_tests/test_interpolation.py
index 63cda967b204d3773241765f3482aaa6421551cb..fbb3067e6a64bf3d43d06536637b9df9be56863a 100644
--- a/pystencils_tests/test_interpolation.py
+++ b/pystencils_tests/test_interpolation.py
@@ -155,7 +155,8 @@ def test_rotate_interpolation_gpu(dtype, address_mode):
         previous_result = out.get()
 
 
-def test_shift_interpolation_gpu():
+@pytest.mark.parametrize('address_mode', ['border', 'wrap', 'clamp', 'mirror'])
+def test_shift_interpolation_gpu(address_mode):
 
     rotation_angle = 0  # sympy.pi / 5
     scale = 1