def test_convert_calendar_missing(source, target, freq):
src = DataArray(
date_range(
"2004-01-01",
"2004-12-31" if source != "360_day" else "2004-12-30",
freq=freq,
calendar=source,
),
dims=("time", ),
name="time",
)
da_src = DataArray(np.linspace(0, 1, src.size),
dims=("time", ),
coords={"time": src})
out = convert_calendar(da_src, target, missing=np.nan, align_on="date")
assert infer_freq(out.time) == freq
expected = date_range(
"2004-01-01",
"2004-12-31" if target != "360_day" else "2004-12-30",
freq=freq,
calendar=target,
)
np.testing.assert_array_equal(out.time, expected)
if freq != "M":
out_without_missing = convert_calendar(da_src, target, align_on="date")
expected_nan = out.isel(time=~out.time.isin(out_without_missing.time))
assert expected_nan.isnull().all()
expected_not_nan = out.sel(time=out_without_missing.time)
assert_identical(expected_not_nan, out_without_missing)
def test_convert_calendar_errors():
src_nl = DataArray(
date_range("0000-01-01", "0000-12-31", freq="D", calendar="noleap"),
dims=("time", ),
name="time",
)
# no align_on for conversion to 360_day
with pytest.raises(ValueError,
match="Argument `align_on` must be specified"):
convert_calendar(src_nl, "360_day")
# Standard doesn't support year 0
with pytest.raises(
ValueError,
match="Source time coordinate contains dates with year 0"):
convert_calendar(src_nl, "standard")
# no align_on for conversion from 360 day
src_360 = convert_calendar(src_nl, "360_day", align_on="year")
with pytest.raises(ValueError,
match="Argument `align_on` must be specified"):
convert_calendar(src_360, "noleap")
# Datetime objects
da = DataArray([0, 1, 2], dims=("x", ), name="x")
with pytest.raises(ValueError,
match="Coordinate x must contain datetime objects."):
convert_calendar(da, "standard", dim="x")
def test_convert_calendar_360_days(source, target, freq, align_on):
src = DataArray(
date_range("2004-01-01", "2004-12-30", freq=freq, calendar=source),
dims=("time", ),
name="time",
)
da_src = DataArray(np.linspace(0, 1, src.size),
dims=("time", ),
coords={"time": src})
conv = convert_calendar(da_src, target, align_on=align_on)
assert conv.time.dt.calendar == target
if align_on == "date":
np.testing.assert_array_equal(
conv.time.resample(time="M").last().dt.day,
[30, 29, 30, 30, 30, 30, 30, 30, 30, 30, 30, 30],
)
elif target == "360_day":
np.testing.assert_array_equal(
conv.time.resample(time="M").last().dt.day,
[30, 30, 30, 30, 30, 30, 30, 30, 30, 30, 30, 29],
)
else:
np.testing.assert_array_equal(
conv.time.resample(time="M").last().dt.day,
[30, 29, 30, 30, 31, 30, 30, 31, 30, 31, 29, 31],
)
if source == "360_day" and align_on == "year":
assert conv.size == 360 if freq == "D" else 360 * 4
else:
assert conv.size == 359 if freq == "D" else 359 * 4
def test_convert_calendar_same_calendar():
src = DataArray(
date_range("2000-01-01", periods=12, freq="6H", use_cftime=False),
dims=("time", ),
name="time",
)
out = convert_calendar(src, "proleptic_gregorian")
assert src is out
def test_convert_calendar(source, target, use_cftime, freq):
src = DataArray(
date_range("2004-01-01", "2004-12-31", freq=freq, calendar=source),
dims=("time", ),
name="time",
)
da_src = DataArray(np.linspace(0, 1, src.size),
dims=("time", ),
coords={"time": src})
conv = convert_calendar(da_src, target, use_cftime=use_cftime)
assert conv.time.dt.calendar == target
if source != "noleap":
expected_times = date_range(
"2004-01-01",
"2004-12-31",
freq=freq,
use_cftime=use_cftime,
calendar=target,
)
else:
expected_times_pre_leap = date_range(
"2004-01-01",
"2004-02-28",
freq=freq,
use_cftime=use_cftime,
calendar=target,
)
expected_times_post_leap = date_range(
"2004-03-01",
"2004-12-31",
freq=freq,
use_cftime=use_cftime,
calendar=target,
)
expected_times = expected_times_pre_leap.append(
expected_times_post_leap)
np.testing.assert_array_equal(conv.time, expected_times)