Skip to content

Commit 6be7500

Browse files
Adding unit test for bug in writing output when time_interval=None
This mimics the bug described in #2382
1 parent 5f2e8ec commit 6be7500

1 file changed

Lines changed: 16 additions & 0 deletions

File tree

tests/test_particlefile.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
)
2222
from parcels._core.particle import Particle, create_particle_data, get_default_particle
2323
from parcels._core.utils.time import TimeInterval
24+
from parcels._datasets.structured.generated import peninsula_dataset
2425
from parcels._datasets.structured.generic import datasets
2526
from parcels.interpolators import XLinear
2627
from parcels.kernels import AdvectionRK4
@@ -69,6 +70,21 @@ def test_pfile_array_write_zarr_memorystore(fieldset):
6970
assert ds.sizes["trajectory"] == npart
7071

7172

73+
def test_write_fieldset_without_time(tmp_zarrfile):
74+
ds = peninsula_dataset() # DataSet without time
75+
assert "time" not in ds.dims
76+
grid = XGrid.from_dataset(ds)
77+
fieldset = FieldSet([Field("U", ds["U"], grid, XLinear)])
78+
79+
pset = ParticleSet(fieldset, pclass=Particle, lon=0, lat=0)
80+
81+
ofile = ParticleFile(tmp_zarrfile, outputdt=np.timedelta64(1, "s"))
82+
pset.execute(DoNothing, runtime=np.timedelta64(1, "s"), dt=np.timedelta64(1, "s"), output_file=ofile)
83+
84+
ds = xr.open_zarr(tmp_zarrfile)
85+
assert ds.time.values[0, 1] == np.timedelta64(1, "s")
86+
87+
7288
def test_pfile_array_remove_particles(fieldset, tmp_zarrfile):
7389
npart = 10
7490
pset = ParticleSet(

0 commit comments

Comments
 (0)