def test_extend_weekend_new_label(self):
amds = {'17 Dec 2017': 5}
result = extend_weekends(amds, how='nearest', label='a')
assert result == {
get_timestamp('17 Dec 2017'): 5,
get_timestamp('18 Dec 2017'): 'a'
}
def test_extend_weekend_sunday(self):
amds = {'17 Dec 2017': 5}
result = extend_weekends(amds, how='nearest')
assert result == {
get_timestamp('17 Dec 2017'): 5,
get_timestamp('18 Dec 2017'): 5
}
result = extend_weekends(amds, how='previous')
assert result == {
get_timestamp('17 Dec 2017'): 5,
get_timestamp('15 Dec 2017'): 5
}
result = extend_weekends(amds, how='next')
assert result == {
get_timestamp('17 Dec 2017'): 5,
get_timestamp('18 Dec 2017'): 5
}
def test_extend_weekend_weekday_already_off2(self):
amds = {'17 Dec 2017': 5, '15 Dec 2017': 5}
result = extend_weekends(amds, how='previous')
assert result == {
get_timestamp('14 Dec 2017'): 5,
get_timestamp('15 Dec 2017'): 5,
get_timestamp('17 Dec 2017'): 5
}
def test_extend_weekend_weekday_already_off1(self):
amds = {'17 Dec 2017': 5, '18 Dec 2017': 5}
result = extend_weekends(amds, how='nearest')
assert result == {
get_timestamp('17 Dec 2017'): 5,
get_timestamp('18 Dec 2017'): 5,
get_timestamp('19 Dec 2017'): 5
}
def test_extend_weekend_strange_weekend(self):
amds = {'16 Dec 2017': 5}
result = extend_weekends(amds, how='nearest', weekend=[5])
# if there is a tie, 'nearest' == 'next'
assert result == {
get_timestamp('16 Dec 2017'): 5,
get_timestamp('17 Dec 2017'): 5
}
result = extend_weekends(amds, how='next', weekend=[5, 6, 0])
assert result == {
get_timestamp('16 Dec 2017'): 5,
get_timestamp('19 Dec 2017'): 5
}
result = extend_weekends(amds, how='previous', weekend=[3, 4, 5])
assert result == {
get_timestamp('13 Dec 2017'): 5,
get_timestamp('16 Dec 2017'): 5
}
result = extend_weekends(amds, weekend=[])
assert result == {get_timestamp('16 Dec 2017'): 5}
def test_extend_weekend_two_holidays_diff_labels(self):
amds = {'16 Dec 2017': 3, '17 Dec 2017': 5}
result = extend_weekends(amds, how='nearest')
assert result == {
get_timestamp('15 Dec 2017'): 3,
get_timestamp('16 Dec 2017'): 3,
get_timestamp('17 Dec 2017'): 5,
get_timestamp('18 Dec 2017'): 5
}
result = extend_weekends(amds, how='previous')
assert result == {
get_timestamp('14 Dec 2017'): 5,
get_timestamp('15 Dec 2017'): 3,
get_timestamp('16 Dec 2017'): 3,
get_timestamp('17 Dec 2017'): 5,
}
result = extend_weekends(amds, how='next')
assert result == {
get_timestamp('16 Dec 2017'): 3,
get_timestamp('17 Dec 2017'): 5,
get_timestamp('18 Dec 2017'): 3,
get_timestamp('19 Dec 2017'): 5
}
def test_extend_weekend_two_holidays_new_label(self):
amds = {'16 Dec 2017': 5, '17 Dec 2017': 5}
result = extend_weekends(amds, how='nearest', label=0)
assert result == {
get_timestamp('15 Dec 2017'): 0,
get_timestamp('16 Dec 2017'): 5,
get_timestamp('17 Dec 2017'): 5,
get_timestamp('18 Dec 2017'): 0
}
result = extend_weekends(amds, how='previous', label=0)
assert result == {
get_timestamp('14 Dec 2017'): 0,
get_timestamp('15 Dec 2017'): 0,
get_timestamp('16 Dec 2017'): 5,
get_timestamp('17 Dec 2017'): 5,
}
result = extend_weekends(amds, how='next', label=0)
assert result == {
get_timestamp('16 Dec 2017'): 5,
get_timestamp('17 Dec 2017'): 5,
get_timestamp('18 Dec 2017'): 0,
get_timestamp('19 Dec 2017'): 0
}
def test_extend_weekend_weekday(self):
amds = {'15 Dec 2017': 5}
result = extend_weekends(amds, how='nearest')
assert result == {get_timestamp('15 Dec 2017'): 5}