Fixing unit test

Author: erikvansebille

tests/test_interpolation.py

@@ -65,11 +65,6 @@ def create_interpolation_data_with_land():
             "lon": np.linspace(0, 1, xdim),
         },
     )
-    # Set a land point (for testing freeslip)
-    ds["U"][:, :, 2, 5] = 0.0
-    ds["V"][:, :, 2, 5] = 0.0
-    ds["W"][:, :, 2, 5] = 0.0
-
     return ds
 
 
@@ -148,8 +143,11 @@ def test_scipy_vs_jit(interp_method):
     dimensions = {"time": "time", "lon": "lon", "lat": "lat", "depth": "depth"}
     fieldset = FieldSet.from_xarray_dataset(create_interpolation_data_with_land(), variables, dimensions, mesh="flat")
 
-    for field in [fieldset.U, fieldset.V, fieldset.W]:  # Set a land point (for testing freeslip)
+    for field in [fieldset.U, fieldset.V, fieldset.W]:
         field.interp_method = interp_method
+        # Set a land point (for testing freeslip)
+        if interp_method == "freeslip":
+            field.data[0, :, 2, 5] = 0
 
     x, y, z = np.meshgrid(np.linspace(0, 1, 7), np.linspace(0, 1, 13), np.linspace(0, 1, 5))
 
@@ -162,7 +160,7 @@ def DeleteParticle(particle, fieldset, time):
             particle.delete()
 
     for pset in [pset_scipy, pset_jit]:
-        pset.execute([AdvectionRK4_3D, DeleteParticle], runtime=4e-3, dt=1e-3)
+        pset.execute([AdvectionRK4_3D, DeleteParticle], runtime=4, dt=1)
 
     tol = 1e-6
     for i in range(len(pset_scipy)):