|
21 | 21 | ) |
22 | 22 | from parcels._core.particle import Particle, create_particle_data, get_default_particle |
23 | 23 | from parcels._core.utils.time import TimeInterval |
| 24 | +from parcels._datasets.structured.generated import peninsula_dataset |
24 | 25 | from parcels._datasets.structured.generic import datasets |
25 | 26 | from parcels.interpolators import XLinear |
26 | 27 | from parcels.kernels import AdvectionRK4 |
@@ -69,6 +70,21 @@ def test_pfile_array_write_zarr_memorystore(fieldset): |
69 | 70 | assert ds.sizes["trajectory"] == npart |
70 | 71 |
|
71 | 72 |
|
| 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 | + |
72 | 88 | def test_pfile_array_remove_particles(fieldset, tmp_zarrfile): |
73 | 89 | npart = 10 |
74 | 90 | pset = ParticleSet( |
|
0 commit comments