def test_cftime_range(
start, end, periods, freq, closed, normalize, calendar,
expected_date_args):
date_type = get_date_type(calendar)
expected_dates = [date_type(*args) for args in expected_date_args]
if isinstance(start, tuple):
start = date_type(*start)
if isinstance(end, tuple):
end = date_type(*end)
result = cftime_range(
start=start, end=end, periods=periods, freq=freq, closed=closed,
normalize=normalize, calendar=calendar)
resulting_dates = result.values
assert isinstance(result, CFTimeIndex)
if freq is not None:
np.testing.assert_equal(resulting_dates, expected_dates)
else:
# If we create a linear range of dates using cftime.num2date
# we will not get exact round number dates. This is because
# datetime arithmetic in cftime is accurate approximately to
# 1 millisecond (see https://unidata.github.io/cftime/api.html).
deltas = resulting_dates - expected_dates
deltas = np.array([delta.total_seconds() for delta in deltas])
assert np.max(np.abs(deltas)) < 0.001
def test_calendar_specific_month_end(freq, calendar, expected_month_day):
year = 2000 # Use a leap-year to highlight calendar differences
result = cftime_range(
start='2000-02', end='2001', freq=freq, calendar=calendar).values
date_type = get_date_type(calendar)
expected = [date_type(year, *args) for args in expected_month_day]
np.testing.assert_equal(result, expected)
def test_to_cftime_datetime(calendar, argument, expected_date_args):
date_type = get_date_type(calendar)
expected = date_type(*expected_date_args)
if isinstance(argument, tuple):
argument = date_type(*argument)
result = to_cftime_datetime(argument, calendar=calendar)
assert result == expected
def test_add_year_begin(calendar, initial_date_args, offset,
expected_date_args):
date_type = get_date_type(calendar)
initial = date_type(*initial_date_args)
result = initial + offset
expected = date_type(*expected_date_args)
assert result == expected
def test_days_in_month_december(calendar):
if calendar == '360_day':
expected = 30
else:
expected = 31
date_type = get_date_type(calendar)
reference = date_type(1, 12, 5)
assert _days_in_month(reference) == expected
def test_onOffset_month_or_year_end(
calendar, year_month_args, sub_day_args, offset):
date_type = get_date_type(calendar)
reference_args = year_month_args + (1,)
reference = date_type(*reference_args)
date_args = year_month_args + (_days_in_month(reference),) + sub_day_args
date = date_type(*date_args)
result = offset.onOffset(date)
assert result
def test_add_year_end(
calendar, initial_date_args, offset, expected_year_month,
expected_sub_day
):
date_type = get_date_type(calendar)
initial = date_type(*initial_date_args)
result = initial + offset
reference_args = expected_year_month + (1,)
reference = date_type(*reference_args)
# Here the days at the end of each month varies based on the calendar used
expected_date_args = (expected_year_month +
(_days_in_month(reference),) + expected_sub_day)
expected = date_type(*expected_date_args)
assert result == expected
def test_add_year_end(
calendar, initial_date_args, offset, expected_year_month,
expected_sub_day
):
date_type = get_date_type(calendar)
initial = date_type(*initial_date_args)
result = initial + offset
reference_args = expected_year_month + (1,)
reference = date_type(*reference_args)
# Here the days at the end of each month varies based on the calendar used
expected_date_args = (expected_year_month +
(_days_in_month(reference),) + expected_sub_day)
expected = date_type(*expected_date_args)
assert result == expected
def test_rollback(calendar, offset, initial_date_args,
partial_expected_date_args):
date_type = get_date_type(calendar)
initial = date_type(*initial_date_args)
if isinstance(offset, (MonthBegin, YearBegin)):
expected_date_args = partial_expected_date_args + (1, )
elif isinstance(offset, (MonthEnd, YearEnd)):
reference_args = partial_expected_date_args + (1, )
reference = date_type(*reference_args)
expected_date_args = (partial_expected_date_args +
(_days_in_month(reference), ))
else:
expected_date_args = partial_expected_date_args
expected = date_type(*expected_date_args)
result = offset.rollback(initial)
assert result == expected
def test_rollback(calendar, offset, initial_date_args,
partial_expected_date_args):
date_type = get_date_type(calendar)
initial = date_type(*initial_date_args)
if isinstance(offset, (MonthBegin, YearBegin)):
expected_date_args = partial_expected_date_args + (1,)
elif isinstance(offset, (MonthEnd, YearEnd)):
reference_args = partial_expected_date_args + (1,)
reference = date_type(*reference_args)
expected_date_args = (partial_expected_date_args +
(_days_in_month(reference),))
else:
expected_date_args = partial_expected_date_args
expected = date_type(*expected_date_args)
result = offset.rollback(initial)
assert result == expected
def test_onOffset(calendar, date_args, offset, expected):
date_type = get_date_type(calendar)
date = date_type(*date_args)
result = offset.onOffset(date)
assert result == expected
def test_onOffset(calendar, date_args, offset, expected):
date_type = get_date_type(calendar)
date = date_type(*date_args)
result = offset.onOffset(date)
assert result == expected
def test_days_in_month_non_december(calendar):
date_type = get_date_type(calendar)
reference = date_type(1, 4, 1)
assert _days_in_month(reference) == 30
def test_sub_error(offset, calendar):
date_type = get_date_type(calendar)
initial = date_type(1, 1, 1)
with pytest.raises(TypeError):
offset - initial
def test_rsub_sub_monthly(offset, expected_date_args, calendar):
date_type = get_date_type(calendar)
initial = date_type(1, 1, 3)
expected = date_type(*expected_date_args)
result = initial - offset
assert result == expected
def test_add_sub_monthly(offset, expected_date_args, calendar):
date_type = get_date_type(calendar)
initial = date_type(1, 1, 1)
expected = date_type(*expected_date_args)
result = offset + initial
assert result == expected
def test_add_sub_monthly(offset, expected_date_args, calendar):
date_type = get_date_type(calendar)
initial = date_type(1, 1, 1)
expected = date_type(*expected_date_args)
result = offset + initial
assert result == expected
def test_rsub_sub_monthly(offset, expected_date_args, calendar):
date_type = get_date_type(calendar)
initial = date_type(1, 1, 3)
expected = date_type(*expected_date_args)
result = initial - offset
assert result == expected
def test_sub_error(offset, calendar):
date_type = get_date_type(calendar)
initial = date_type(1, 1, 1)
with pytest.raises(TypeError):
offset - initial
def test_add_quarter_begin(calendar, initial_date_args, offset, expected_date_args):
date_type = get_date_type(calendar)
initial = date_type(*initial_date_args)
result = initial + offset
expected = date_type(*expected_date_args)
assert result == expected
def test_days_in_month_non_december(calendar):
date_type = get_date_type(calendar)
reference = date_type(1, 4, 1)
assert _days_in_month(reference) == 30
