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))