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_interp_calendar_errors():
src_nl = DataArray(
[1] * 100,
dims=("time", ),
coords={
"time":
date_range("0000-01-01", periods=100, freq="MS", calendar="noleap")
},
)
tgt_360 = date_range("0001-01-01",
"0001-12-30",
freq="MS",
calendar="standard")
with pytest.raises(
ValueError,
match="Source time coordinate contains dates with year 0"):
interp_calendar(src_nl, tgt_360)
da1 = DataArray([0, 1, 2], dims=("x", ), name="x")
da2 = da1 + 1
with pytest.raises(
ValueError,
match="Both 'source.x' and 'target' must contain datetime objects."
):
interp_calendar(da1, da2, 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_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_date_range_like_errors():
src = date_range("1899-02-03", periods=20, freq="D", use_cftime=False)
src = src[np.arange(20) != 10] # Remove 1 day so the frequency is not inferable.
with pytest.raises(
ValueError,
match="`date_range_like` was unable to generate a range as the source frequency was not inferable.",
):
date_range_like(src, "gregorian")
src = DataArray(
np.array(
[["1999-01-01", "1999-01-02"], ["1999-01-03", "1999-01-04"]],
dtype=np.datetime64,
),
dims=("x", "y"),
)
with pytest.raises(
ValueError,
match="'source' must be a 1D array of datetime objects for inferring its range.",
):
date_range_like(src, "noleap")
da = DataArray([1, 2, 3, 4], dims=("time",))
with pytest.raises(
ValueError,
match="'source' must be a 1D array of datetime objects for inferring its range.",
):
date_range_like(da, "noleap")
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_date_range(start, calendar, use_cftime, expected_type):
dr = date_range(start,
periods=14,
freq="D",
calendar=calendar,
use_cftime=use_cftime)
assert isinstance(dr, expected_type)
def test_interp_calendar(source, target):
src = DataArray(
date_range("2004-01-01", "2004-07-30", freq="D", calendar=source),
dims=("time", ),
name="time",
)
tgt = DataArray(
date_range("2004-01-01", "2004-07-30", freq="D", calendar=target),
dims=("time", ),
name="time",
)
da_src = DataArray(np.linspace(0, 1, src.size),
dims=("time", ),
coords={"time": src})
conv = interp_calendar(da_src, tgt)
assert_identical(tgt.time, conv.time)
np.testing.assert_almost_equal(conv.max(), 1, 2)
assert conv.min() == 0
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)
def test_date_range_like(start, freq, cal_src, cal_tgt, use_cftime, exp0, exp_pd):
source = date_range(start, periods=12, freq=freq, calendar=cal_src)
out = date_range_like(source, cal_tgt, use_cftime=use_cftime)
assert len(out) == 12
assert infer_freq(out) == freq
assert out[0].isoformat().startswith(exp0)
if exp_pd:
assert isinstance(out, pd.DatetimeIndex)
else:
assert isinstance(out, CFTimeIndex)
assert out.calendar == cal_tgt
def test_date_range_errors():
with pytest.raises(ValueError, match="Date range is invalid"):
date_range(
"1400-01-01", periods=1, freq="D", calendar="standard", use_cftime=False
)
with pytest.raises(ValueError, match="Date range is invalid"):
date_range(
"2480-01-01",
periods=1,
freq="D",
calendar="proleptic_gregorian",
use_cftime=False,
)
with pytest.raises(ValueError, match="Invalid calendar "):
date_range(
"1900-01-01", periods=1, freq="D", calendar="noleap", use_cftime=False
)
def test_date_range_like_same_calendar():
src = date_range("2000-01-01", periods=12, freq="6H", use_cftime=False)
out = date_range_like(src, "standard", use_cftime=False)
assert src is out
