Adding one extra assert to check dataset and dict equivalence

Author: erikvansebille

tests/v4/test_particleset_execute.py

@@ -116,7 +116,7 @@ def PythonFail(particle, fieldset, time):  # pragma: no cover
     assert all([time == fieldset.time_interval.left + np.timedelta64(0, "s") for time in pset.time[1:]])
 
 
-def test_pset_update_particle(fieldset, npart=10):
+def test_pset_update_particles_in_dataset_and_dict(fieldset, npart=10):
     lon_start = np.linspace(0, 1, npart)
     lat_start = np.linspace(1, 0, npart)
     pset = ParticleSet(fieldset, lon=np.linspace(0, 1, npart), lat=np.linspace(1, 0, npart))
@@ -128,6 +128,7 @@ def UpdateParticle(particle, fieldset, time):  # pragma: no cover
     pset.execute(pset.Kernel(UpdateParticle), runtime=np.timedelta64(10, "s"), dt=np.timedelta64(1, "s"))
     assert np.allclose(pset.lon, lon_start + 1, atol=1e-5)
     assert np.allclose(pset.lat, lat_start - 1, atol=1e-5)
+    assert all(pset._data["lon"] == pset._ds["lon"].data)
 
 
 @pytest.mark.parametrize("verbose_progress", [True, False])