Adding more tests

Author: erikvansebille

tests/v4/test_particleset.py

@@ -16,6 +16,7 @@
 )
 from parcels._datasets.structured.generic import datasets as datasets_structured
 from parcels.xgrid import XGrid
+from tests.common_kernels import DoNothing
 
 
 @pytest.fixture
@@ -27,10 +28,6 @@ def fieldset() -> FieldSet:
     return FieldSet([U, V])
 
 
-def DoNothing(particle, fieldset, time):
-    pass
-
-
 def test_pset_create_lon_lat(fieldset):
     npart = 100
     lon = np.linspace(0, 1, npart, dtype=np.float32)

tests/v4/test_particleset_execute.py

@@ -16,6 +16,7 @@
 from parcels._datasets.unstructured.generic import datasets as datasets_unstructured
 from parcels.uxgrid import UxGrid
 from parcels.xgrid import XGrid
+from tests.common_kernels import DoNothing
 
 
 @pytest.fixture
@@ -67,6 +68,25 @@ def AddLat(particle, fieldset, time):  # pragma: no cover
         assert np.allclose([p.lat - n * 0.1 for p in pset], np.zeros(npart), rtol=1e-12)
 
 
+@pytest.mark.parametrize(
+    "starttime, endtime, dt",
+    [
+        (0, 10, 1),
+        (0, 10, 3),
+        (2, 16, 3),
+        (20, 10, -1),
+        (20, -10, -2),
+    ],
+)
+def test_execution_endtime(fieldset, starttime, endtime, dt):
+    starttime = fieldset.time_interval.left + np.timedelta64(starttime, "s")
+    endtime = fieldset.time_interval.left + np.timedelta64(endtime, "s")
+    dt = np.timedelta64(dt, "s")
+    pset = ParticleSet(fieldset, time=starttime, lon=0, lat=0)
+    pset.execute(DoNothing, endtime=endtime, dt=dt)
+    assert np.isclose(pset.time_nextloop.values, endtime)
+
+
 @pytest.mark.parametrize("verbose_progress", [True, False])
 def test_uxstommelgyre_pset_execute(verbose_progress):
     ds = datasets_unstructured["stommel_gyre_delaunay"]