def test_reject_timezoned_timestamp_without_calling_individual_function(
self):
tz = pytz.timezone("Europe/Berlin")
dt = tz.normalize(datetime.utcnow().replace(tzinfo=pytz.UTC))
with contextlib.ExitStack() as stack:
align_to_days = stack.enter_context(
unittest.mock.patch("cuteborg.scheduler.utils.align_to_days"))
stack.enter_context(
self.assertRaisesRegex(ValueError, "needs naive datetime"))
utils.align_to_interval(
dt,
2,
config.IntervalUnit.DAY,
)
align_to_days.assert_not_called()
def test_dispatch_to_year_impl(self):
with contextlib.ExitStack() as stack:
align_to_years = stack.enter_context(
unittest.mock.patch(
"cuteborg.scheduler.utils.align_to_years", ))
result = utils.align_to_interval(
self.dt,
unittest.mock.sentinel.step,
config.IntervalUnit.YEAR,
)
align_to_years.assert_called_once_with(
self.dt,
unittest.mock.sentinel.step,
)
self.assertEqual(
result,
align_to_years(),
)