Skip to content

Commit 77d961a

Browse files
Simplify datetime64 dt.calendar tests (#6988)
1 parent 5bec466 commit 77d961a

File tree

1 file changed

+4
-9
lines changed

1 file changed

+4
-9
lines changed

xarray/tests/test_accessor_dt.py

Lines changed: 4 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -423,22 +423,17 @@ def test_calendar_cftime(data) -> None:
423423

424424

425425
@requires_cftime
426-
def test_calendar_cftime_2D(data) -> None:
427-
# 2D np datetime:
428-
data = xr.DataArray(
429-
np.random.randint(1, 1000000, size=(4, 5)).astype("<M8[h]"), dims=("x", "y")
430-
)
426+
def test_calendar_datetime64_2d() -> None:
427+
data = xr.DataArray(np.zeros((4, 5), dtype="datetime64[ns]"), dims=("x", "y"))
431428
assert data.dt.calendar == "proleptic_gregorian"
432429

433430

434431
@requires_dask
435-
def test_calendar_dask() -> None:
432+
def test_calendar_datetime64_3d_dask() -> None:
436433
import dask.array as da
437434

438-
# 3D lazy dask - np
439435
data = xr.DataArray(
440-
da.random.randint(1, 1000000 + 1, size=(4, 5, 6)).astype("<M8[h]"),
441-
dims=("x", "y", "z"),
436+
da.zeros((4, 5, 6), dtype="datetime64[ns]"), dims=("x", "y", "z")
442437
)
443438
with raise_if_dask_computes():
444439
assert data.dt.calendar == "proleptic_gregorian"

0 commit comments

Comments
 (0)