fixing unit tests for new kernel loop

Author: erikvansebille

src/parcels/_core/index_search.py

@@ -85,8 +85,8 @@ def _search_time_index(field: Field, time: datetime):
     if not field.time_interval.is_all_time_in_interval(time):
         _raise_outside_time_interval_error(time, field=None)
 
-    ti = np.searchsorted(field.data.time.data, time, side="right") - 1
-    tau = (time - field.data.time.data[ti]) / (field.data.time.data[ti + 1] - field.data.time.data[ti])
+    ti, tau = _search_1d_array(field.data.time.data, time)
+
     return {"T": {"index": np.atleast_1d(ti), "bcoord": np.atleast_1d(tau)}}
 
 

tests/test_particleset_execute.py

@@ -159,18 +159,21 @@ def SampleU(particles, fieldset):  # pragma: no cover
 
     pset = ParticleSet(fieldset, lon=[0.2], lat=[5.0], time=[starttime])
     pset.execute(SampleU, endtime=endtime, dt=np.timedelta64(1, "D"))
-    assert pset[0].time + pset[0].dt == endtime
+    assert pset[0].time == endtime
 
 
 def test_particleset_interpolate_on_domainedge(zonal_flow_fieldset):
     fieldset = zonal_flow_fieldset
 
+    MyParticle = Particle.add_variable(Variable("var"))
+
     def SampleU(particles, fieldset):  # pragma: no cover
-        particles.dlon = fieldset.U[particles]
+        particles.var = fieldset.U[particles]
 
-    pset = ParticleSet(fieldset, lon=fieldset.U.grid.lon[-1], lat=fieldset.U.grid.lat[-1])
+    print(fieldset.U.grid.lon)
+    pset = ParticleSet(fieldset, pclass=MyParticle, lon=fieldset.U.grid.lon[-1], lat=fieldset.U.grid.lat[-1])
     pset.execute(SampleU, runtime=np.timedelta64(1, "D"), dt=np.timedelta64(1, "D"))
-    np.testing.assert_equal(pset[0].dlon, 1)
+    np.testing.assert_equal(pset[0].var, 1)
 
 
 def test_particleset_interpolate_outside_domainedge(zonal_flow_fieldset):