Skip to content

Commit 3693329

Browse files
committed
Fix test_particle_schema
1 parent 8626d48 commit 3693329

1 file changed

Lines changed: 7 additions & 3 deletions

File tree

tests/test_particlefile.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424
)
2525
from parcels._core.particle import Particle, get_default_particle
2626
from parcels._core.particlefile import _get_schema
27-
from parcels._core.utils.time import timedelta_to_float
27+
from parcels._core.utils.time import TimeInterval, timedelta_to_float
2828
from parcels._datasets.structured.generated import peninsula_dataset
2929
from parcels._datasets.structured.generic import datasets
3030
from parcels.convert import copernicusmarine_to_sgrid
@@ -496,7 +496,7 @@ def Update_lon(particles, fieldset): # pragma: no cover
496496
],
497497
)
498498
def test_particle_schema(particle):
499-
s = _get_schema(particle, {})
499+
s = _get_schema(particle, {}, TimeInterval(datetime(2023, 1, 1, 12, 0), datetime(2023, 1, 2, 12, 0)))
500500

501501
written_variables = [v for v in particle.variables if v.to_write]
502502

@@ -510,5 +510,9 @@ def test_particle_schema(particle):
510510
strict=False,
511511
):
512512
assert variable.name == pyarrow_field.name
513-
assert variable.attrs == {k.decode(): v.decode() for k, v in pyarrow_field.metadata.items()}
513+
if variable.name != "time":
514+
assert variable.attrs == {k.decode(): v.decode() for k, v in pyarrow_field.metadata.items()}
515+
else:
516+
assert b"units" in pyarrow_field.metadata
517+
assert b"calendar" in pyarrow_field.metadata
514518
assert pa.from_numpy_dtype(variable.dtype) == pyarrow_field.type

0 commit comments

Comments
 (0)