def test_daily_e_s_t(self):
self.tz = tz.gettz("UTC")
dt = datetime(2020, 2, 2, 10, 0, 0, tzinfo=self.tz)
start, end, increment = date_helper.get_session_times(
dt, self.tz, "09:00:00", "08:00:00")
self.assertEqual(dt.date(), start.date())
self.assertEqual(dt.date(), end.date() - timedelta(days=1))
self.assertEqual(increment, timedelta(days=1))
def test_weekly_t_s_e(self):
self.tz = tz.gettz("UTC")
dt = datetime(2020, 2, 4, 10, 0, 0, tzinfo=self.tz)
start, end, increment = date_helper.get_session_times(
dt, self.tz, "Wednesday 11:00:00", "Friday 12:00:00")
self.assertEqual(start, datetime(2020, 2, 5, 11, 0, 0, tzinfo=self.tz))
self.assertEqual(end, datetime(2020, 2, 7, 12, 0, 0, tzinfo=self.tz))
self.assertEqual(increment, timedelta(days=7))
def test_daily_t_e_s(self):
self.tz = tz.gettz("UTC")
dt = datetime(2020, 2, 2, 10, 0, 0, tzinfo=self.tz)
start, end, increment = date_helper.get_session_times(
dt, self.tz, "12:00:00", "11:00:00")
self.assertEqual(start, datetime(2020, 2, 1, 12, 0, 0, tzinfo=self.tz))
self.assertEqual(end, datetime(2020, 2, 2, 11, 0, 0, tzinfo=self.tz))
self.assertEqual(increment, timedelta(days=1))