def test_extract_months_single_month():
time = xr.DataArray(pd.date_range(start='1678-01-01', end='1678-01-31',
freq='1M'), dims=[TIME_STR])
months = 1
desired = time
actual = extract_months(time, months)
xr.testing.assert_identical(actual, desired)
def test_extract_months_single_month():
time = xr.DataArray(pd.date_range(start='1678-01-01',
end='1678-01-31',
freq='1M'),
dims=[TIME_STR])
months = 1
desired = time
actual = extract_months(time, months)
xr.testing.assert_identical(actual, desired)
def test_extract_months():
time = xr.DataArray(pd.date_range(start='2001-02-18', end='2002-07-12',
freq='1D'), dims=[TIME_STR])
months = 'mam' # March-April-May
desired = xr.concat([
xr.DataArray(pd.date_range(start='2001-03-01', end='2001-05-31',
freq='1D'), dims=[TIME_STR]),
xr.DataArray(pd.date_range(start='2002-03-01', end='2002-05-31',
freq='1D'), dims=[TIME_STR])
], dim=TIME_STR)
actual = extract_months(time, months)
xr.testing.assert_identical(actual, desired)
def test_extract_months():
time = xr.DataArray(pd.date_range(start='2001-02-18',
end='2002-07-12',
freq='1D'),
dims=[TIME_STR])
months = 'mam' # March-April-May
desired = xr.concat([
xr.DataArray(pd.date_range(
start='2001-03-01', end='2001-05-31', freq='1D'),
dims=[TIME_STR]),
xr.DataArray(pd.date_range(
start='2002-03-01', end='2002-05-31', freq='1D'),
dims=[TIME_STR])
],
dim=TIME_STR)
actual = extract_months(time, months)
xr.testing.assert_identical(actual, desired)