def test_reset_end(self):
daily_rule = turoboro.DailyRule(datetime(2014, 1, 1), datetime(2014, 5, 30))
self.assertEqual(daily_rule.spec['end'], '2014-05-31T00:00:00+00:00')
daily_rule.end_on(None)
self.assertEqual(daily_rule.spec['end'], None)
daily_rule = turoboro.DailyRule(datetime(2014, 1, 1), datetime(2014, 5, 30))
daily_rule._end_before(None)
self.assertEqual(daily_rule.spec['end'], None)
def test_staggered(self):
daily_rule = turoboro.DailyRule(
datetime(2014, 1, 1), repeat_n_times=44, every_nth_day=7, except_weekdays=turoboro.WEEKEND,
except_months=(turoboro.FEBRUARY, turoboro.OCTOBER), on_hour=14)
expected_all = ['2014-01-01T14:00:00+00:00', '2014-01-08T14:00:00+00:00', '2014-01-15T14:00:00+00:00', '2014-01-22T14:00:00+00:00',
'2014-01-29T14:00:00+00:00', '2014-03-05T14:00:00+00:00', '2014-03-12T14:00:00+00:00', '2014-03-19T14:00:00+00:00',
'2014-03-26T14:00:00+00:00', '2014-04-02T14:00:00+00:00', '2014-04-09T14:00:00+00:00', '2014-04-16T14:00:00+00:00',
'2014-04-23T14:00:00+00:00', '2014-04-30T14:00:00+00:00', '2014-05-07T14:00:00+00:00', '2014-05-14T14:00:00+00:00',
'2014-05-21T14:00:00+00:00', '2014-05-28T14:00:00+00:00', '2014-06-04T14:00:00+00:00', '2014-06-11T14:00:00+00:00',
'2014-06-18T14:00:00+00:00', '2014-06-25T14:00:00+00:00', '2014-07-02T14:00:00+00:00', '2014-07-09T14:00:00+00:00',
'2014-07-16T14:00:00+00:00', '2014-07-23T14:00:00+00:00', '2014-07-30T14:00:00+00:00', '2014-08-06T14:00:00+00:00',
'2014-08-13T14:00:00+00:00', '2014-08-20T14:00:00+00:00', '2014-08-27T14:00:00+00:00', '2014-09-03T14:00:00+00:00',
'2014-09-10T14:00:00+00:00', '2014-09-17T14:00:00+00:00', '2014-09-24T14:00:00+00:00', '2014-11-05T14:00:00+00:00',
'2014-11-12T14:00:00+00:00', '2014-11-19T14:00:00+00:00', '2014-11-26T14:00:00+00:00', '2014-12-03T14:00:00+00:00',
'2014-12-10T14:00:00+00:00', '2014-12-17T14:00:00+00:00', '2014-12-24T14:00:00+00:00', '2014-12-31T14:00:00+00:00']
result = daily_rule.compute()
self.assertEqual(expected_all, result.all)
result_staggered = daily_rule.compute(from_dt=datetime(2014, 5, 29))
self.assertEqual(result_staggered.count, 26)
self.assertEqual(result_staggered.first, '2014-06-04T14:00:00+00:00')
self.assertEqual(result_staggered.last, '2014-12-31T14:00:00+00:00')
result_staggered = daily_rule.compute(from_dt=datetime(2014, 11, 23))
self.assertEqual(result_staggered.first, '2014-11-26T14:00:00+00:00')
self.assertEqual(result_staggered.last, '2014-12-31T14:00:00+00:00')
self.assertEqual(result_staggered.count, 6)
def test(self):
daily_rule = turoboro.DailyRule(datetime(2014, 1, 1), repeat_n_times=10, except_weekdays=turoboro.WEEKEND,
on_hour=8)
expected = ['2014-01-01T08:00:00+00:00', '2014-01-02T08:00:00+00:00', '2014-01-03T08:00:00+00:00', '2014-01-06T08:00:00+00:00',
'2014-01-07T08:00:00+00:00', '2014-01-08T08:00:00+00:00', '2014-01-09T08:00:00+00:00', '2014-01-10T08:00:00+00:00',
'2014-01-13T08:00:00+00:00', '2014-01-14T08:00:00+00:00']
result = daily_rule.compute()
self.assertEqual(result.first, '2014-01-01T08:00:00+00:00')
self.assertEqual(result.last, '2014-01-14T08:00:00+00:00')
self.assertEqual(result.count, 10)
self.assertEqual(expected, result.all)
# Lets change to every third day, until March 31th 2014, excepting weekends and February
daily_rule.every_nth_day(3).except_months(turoboro.FEBRUARY).repeat_n_times(15)
expected_all = ['2014-01-01T08:00:00+00:00', '2014-01-07T08:00:00+00:00', '2014-01-10T08:00:00+00:00', '2014-01-13T08:00:00+00:00',
'2014-01-16T08:00:00+00:00', '2014-01-22T08:00:00+00:00', '2014-01-28T08:00:00+00:00', '2014-01-31T08:00:00+00:00',
'2014-03-05T08:00:00+00:00', '2014-03-11T08:00:00+00:00', '2014-03-14T08:00:00+00:00', '2014-03-17T08:00:00+00:00',
'2014-03-20T08:00:00+00:00', '2014-03-26T08:00:00+00:00', '2014-04-01T08:00:00+00:00']
result = daily_rule.compute()
self.assertEqual(expected_all, result.all)
# Lets get a segment out of our result
expected_segment = ['2014-01-31T08:00:00+00:00', '2014-03-05T08:00:00+00:00', '2014-03-11T08:00:00+00:00', '2014-03-14T08:00:00+00:00',
'2014-03-17T08:00:00+00:00', '2014-03-20T08:00:00+00:00', '2014-03-26T08:00:00+00:00', '2014-04-01T08:00:00+00:00']
self.assertEqual(result.segment(datetime(2014, 1, 31)), expected_segment)
self.assertEqual(result.segment(datetime(2014, 4, 2)), [])
expected_segment = ['2014-01-22T08:00:00+00:00', '2014-01-28T08:00:00+00:00', '2014-01-31T08:00:00+00:00',
'2014-03-05T08:00:00+00:00', '2014-03-11T08:00:00+00:00']
self.assertEqual(result.segment(datetime(2014, 1, 20), datetime(2014, 3, 12)), expected_segment)
def test_kathmandu(self):
daily_rule = turoboro.DailyRule(datetime(2014, 1, 1), except_weekdays=turoboro.WEEKEND, timezone='Asia/Kathmandu')
daily_rule.end_on(datetime(2014, 1, 31))
daily_rule.on_hour(8)
result = daily_rule.compute()
self.assertEqual(result.count, 23)
self.assertEqual(result.first, '2014-01-01T02:15:00+00:00')
def test_next(self):
daily_rule = turoboro.DailyRule(
datetime(2014, 1, 1), repeat_n_times=44, every_nth_day=7, except_weekdays=turoboro.WEEKEND,
except_months=(turoboro.FEBRUARY, turoboro.OCTOBER), on_hour=14
)
result = daily_rule.result()
self.assertEqual(next(result), '2014-01-01T14:00:00+00:00')
self.assertEqual(next(result), '2014-01-08T14:00:00+00:00')
def test_half_year(self):
daily_rule = turoboro.DailyRule(start=datetime(2014, 1, 1), end_on=datetime(2020, 12, 31), every_nth_day=183,
except_weekdays=turoboro.WEEKEND)
expected_half_year_results = ['2014-01-01T00:00:00+00:00', '2014-07-03T00:00:00+00:00', '2015-01-02T00:00:00+00:00',
'2016-07-04T00:00:00+00:00', '2017-01-03T00:00:00+00:00', '2017-07-05T00:00:00+00:00',
'2018-01-04T00:00:00+00:00', '2018-07-06T00:00:00+00:00', '2020-01-06T00:00:00+00:00',
'2020-07-07T00:00:00+00:00']
self.assertEqual(expected_half_year_results, daily_rule.compute().all)
def test_next(self):
daily_rule = turoboro.DailyRule(
datetime(2014, 1, 1), every_nth_day=7, except_weekdays=turoboro.WEEKEND,
except_months=(turoboro.FEBRUARY, turoboro.OCTOBER), on_hour=14
)
result = daily_rule.result(datetime.utcnow())
for res in result:
daily_rule._is_allowed(turoboro.common.datetime_from_isoformat(res))
def test(self):
daily_rule = turoboro.DailyRule(datetime(2014, 1, 1), every_nth_day=5, except_weekdays=turoboro.WEEKEND,
on_hour=8)
result = daily_rule.compute()
self.assertTrue(result.infinite)
self.assertEqual(result.count, 100)
self.assertEqual(result.first, '2014-01-01T08:00:00+00:00')
self.assertEqual(result.last, '2015-11-27T08:00:00+00:00')
result = daily_rule.compute(datetime(2015, 11, 27), max_count_if_infinite=20)
self.assertTrue(result.infinite)
self.assertEqual(result.count, 20)
self.assertEqual(result.first, '2015-12-02T08:00:00+00:00')
self.assertEqual(result.last, '2016-04-15T08:00:00+00:00')
def test(self):
daily_rule = turoboro.DailyRule(datetime(2014, 1, 1), except_weekdays=turoboro.WEEKEND)
# 1. Lets try a daily rule that bypasses weekends for the duration of January 2014
daily_rule.end_on(datetime(2014, 1, 31))
result = daily_rule.compute()
# 2. There are 23 days that do not fall on a weekend in January 2014
self.assertEqual(result.count, 23)
self.assertEqual(result.first, '2014-01-01T00:00:00+00:00')
# 3. Lets set a specific hour and make sure it pans out
daily_rule.on_hour(8)
result = daily_rule.compute()
self.assertEqual(result.first, '2014-01-01T08:00:00+00:00')
self.assertEqual(result.last, '2014-01-31T08:00:00+00:00')
expected_all = ['2014-01-01T08:00:00+00:00', '2014-01-02T08:00:00+00:00', '2014-01-03T08:00:00+00:00', '2014-01-06T08:00:00+00:00',
'2014-01-07T08:00:00+00:00', '2014-01-08T08:00:00+00:00', '2014-01-09T08:00:00+00:00', '2014-01-10T08:00:00+00:00',
'2014-01-13T08:00:00+00:00', '2014-01-14T08:00:00+00:00', '2014-01-15T08:00:00+00:00', '2014-01-16T08:00:00+00:00',
'2014-01-17T08:00:00+00:00', '2014-01-20T08:00:00+00:00', '2014-01-21T08:00:00+00:00', '2014-01-22T08:00:00+00:00',
'2014-01-23T08:00:00+00:00', '2014-01-24T08:00:00+00:00', '2014-01-27T08:00:00+00:00', '2014-01-28T08:00:00+00:00',
'2014-01-29T08:00:00+00:00', '2014-01-30T08:00:00+00:00', '2014-01-31T08:00:00+00:00']
self.assertEqual(result.all, expected_all)
# 4. Lets try with timestamps instead
expected_all_posix = [1388563200, 1388649600, 1388736000, 1388995200, 1389081600, 1389168000, 1389254400,
1389340800, 1389600000, 1389686400, 1389772800, 1389859200, 1389945600, 1390204800,
1390291200, 1390377600, 1390464000, 1390550400, 1390809600, 1390896000, 1390982400,
1391068800, 1391155200]
result = daily_rule.compute(return_as=turoboro.POSIX)
self.assertEqual(result.all, expected_all_posix)
# 5. Lets change to every third day, until March 31th 2014, excepting weekends and February
daily_rule.every_nth_day(3).except_months(turoboro.FEBRUARY).end_on(datetime(2014, 3, 31))
expected_all = ['2014-01-01T08:00:00+00:00', '2014-01-07T08:00:00+00:00', '2014-01-10T08:00:00+00:00', '2014-01-13T08:00:00+00:00',
'2014-01-16T08:00:00+00:00', '2014-01-22T08:00:00+00:00', '2014-01-28T08:00:00+00:00', '2014-01-31T08:00:00+00:00',
'2014-03-05T08:00:00+00:00', '2014-03-11T08:00:00+00:00', '2014-03-14T08:00:00+00:00', '2014-03-17T08:00:00+00:00',
'2014-03-20T08:00:00+00:00', '2014-03-26T08:00:00+00:00']
result = daily_rule.compute()
self.assertEqual(result.all, expected_all)
# 6. Lets get a segment out of our result
expected_segment = ['2014-01-31T08:00:00+00:00', '2014-03-05T08:00:00+00:00', '2014-03-11T08:00:00+00:00', '2014-03-14T08:00:00+00:00',
'2014-03-17T08:00:00+00:00', '2014-03-20T08:00:00+00:00', '2014-03-26T08:00:00+00:00']
self.assertEqual(result.segment(datetime(2014, 1, 31)), expected_segment)
self.assertEqual(result.segment(datetime(2014, 3, 27)), [])
expected_segment = ['2014-01-22T08:00:00+00:00', '2014-01-28T08:00:00+00:00', '2014-01-31T08:00:00+00:00',
'2014-03-05T08:00:00+00:00', '2014-03-11T08:00:00+00:00']
self.assertEqual(result.segment(datetime(2014, 1, 20), datetime(2014, 3, 12)), expected_segment)
def setUp(self):
self.starting_point = datetime(2014, 1, 1)
self.daily_rule = turoboro.DailyRule(self.starting_point)
def test_reset_except_months(self):
daily_rule = turoboro.DailyRule(datetime(2014, 1, 1), except_months=(turoboro.FEBRUARY, turoboro.MARCH))
self.assertEqual(daily_rule.spec['except_months'], (turoboro.FEBRUARY, turoboro.MARCH))
daily_rule.except_months(None)
self.assertEqual(daily_rule.spec['except_months'], None)
def test_reset_except_weekdays(self):
daily_rule = turoboro.DailyRule(datetime(2014, 1, 1), except_weekdays=turoboro.WEEKEND)
self.assertEqual(daily_rule.spec['except_days'], turoboro.WEEKEND)
daily_rule.except_weekdays(None)
self.assertEqual(daily_rule.spec['except_days'], None)