Skip to content

Commit c821248

Browse files
Fixing another particlefile unit test
1 parent 65a7bff commit c821248

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

tests/test_particlefile.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -275,20 +275,19 @@ def SampleP(particles, fieldset): # pragma: no cover
275275
assert fieldset.U.grid.lat[yi] <= lat < fieldset.U.grid.lat[yi + 1]
276276

277277

278-
@pytest.mark.skip
279-
@pytest.mark.v4alpha
280278
def test_reset_dt(fieldset, tmp_zarrfile):
281279
# Assert that p.dt gets reset when a write_time is not a multiple of dt
282280
# for p.dt=0.02 to reach outputdt=0.05 and endtime=0.1, the steps should be [0.2, 0.2, 0.1, 0.2, 0.2, 0.1], resulting in 6 kernel executions
281+
dt = np.timedelta64(20, "ms")
283282

284283
def Update_lon(particles, fieldset): # pragma: no cover
285284
particles.dlon += 0.1
286285

287286
particle = get_default_particle(np.float64)
288287
pset = ParticleSet(fieldset, pclass=particle, lon=[0], lat=[0])
288+
pset.update_dt_dtype(dt.dtype)
289289
ofile = ParticleFile(tmp_zarrfile, outputdt=np.timedelta64(50, "ms"))
290-
dt = np.timedelta64(20, "ms")
291-
pset.execute(pset.Kernel(Update_lon), runtime=6 * dt, dt=dt, output_file=ofile)
290+
pset.execute(pset.Kernel(Update_lon), runtime=5 * dt, dt=dt, output_file=ofile)
292291

293292
assert np.allclose(pset.lon, 0.6)
294293

0 commit comments

Comments
 (0)