def test_working_datetime_ranges_of_date_merging_tomorrow(self):
d = date(year=1993, month=9, day=26)
swh = {
d: [
(time(hour=9), time(hour=12)),
(time(hour=21), time(hour=0)),
]
}
wwh_contiguous = {
tomorrow(d).weekday(): [
(time(hour=0), time(hour=12))
]
}
wwh_separated = {
tomorrow(d).weekday(): [
(time(hour=1), time(hour=12))
]
}
r_s = working_datetime_ranges_of_date(d, swh, wwh_separated,
merge_tomorrow=False)
r_s_merged = working_datetime_ranges_of_date(d, swh, wwh_separated,
merge_tomorrow=True)
r_c = working_datetime_ranges_of_date(d, swh, wwh_contiguous,
merge_tomorrow=False)
r_c_merged = working_datetime_ranges_of_date(d, swh, wwh_contiguous,
merge_tomorrow=True)
expected_r_only_today = [
(
datetime(year=1993, month=9, day=26, hour=9),
datetime(year=1993, month=9, day=26, hour=12),
),
(
datetime(year=1993, month=9, day=26, hour=21),
datetime(year=1993, month=9, day=27, hour=0),
)
]
expected_r_with_tomorrow = [
(
datetime(year=1993, month=9, day=26, hour=9),
datetime(year=1993, month=9, day=26, hour=12),
),
(
datetime(year=1993, month=9, day=26, hour=21),
datetime(year=1993, month=9, day=27, hour=12),
)
]
self.assertEqual(r_s, expected_r_only_today)
self.assertEqual(r_s_merged, expected_r_only_today)
self.assertEqual(r_c, expected_r_only_today)
self.assertEqual(r_c_merged, expected_r_with_tomorrow)
def test_working_datetime_ranges_of_date(self):
d = date(year=1993, month=9, day=26)
swh = {
d: [
(time(hour=9), time(hour=12)),
(time(hour=21), time(hour=0)),
]
}
expected_r = [
(
datetime(year=1993, month=9, day=26, hour=9),
datetime(year=1993, month=9, day=26, hour=12),
),
(
datetime(year=1993, month=9, day=26, hour=21),
datetime(year=1993, month=9, day=27, hour=0),
)
]
self.assertEqual(working_datetime_ranges_of_date(d, swh, {}, True),
expected_r)
self.assertEqual(working_datetime_ranges_of_date(d, swh, {}, False),
expected_r)
def test_working_datetime_ranges_of_date_24_7(self):
d = date(year=2016, month=2, day=18)
wwh_24_7 = {
0: [(time(hour=0), time(hour=0))],
1: [(time(hour=0), time(hour=0))],
2: [(time(hour=0), time(hour=0))],
3: [(time(hour=0), time(hour=0))],
4: [(time(hour=0), time(hour=0))],
5: [(time(hour=0), time(hour=0))],
6: [(time(hour=0), time(hour=0))]
}
expected_ranges = [
(
datetime(year=2016, month=2, day=18, hour=0),
datetime(year=2016, month=2, day=20, hour=0),
)
]
ranges = working_datetime_ranges_of_date(
d, week_working_hours=wwh_24_7, merge_tomorrow=True)
self.assertEqual(ranges, expected_ranges)
def test_working_datetime_ranges_of_date_empty(self):
d = date(year=1993, month=9, day=26)
self.assertEqual(working_datetime_ranges_of_date(d, {}, {}, True), [])
self.assertEqual(working_datetime_ranges_of_date(d, {}, {}, False), [])
