def test_next_week():
expected_range = [
datetime.date(2017, 10, 2),
datetime.date(2017, 10, 3),
datetime.date(2017, 10, 4),
datetime.date(2017, 10, 5),
datetime.date(2017, 10, 6),
datetime.date(2017, 10, 7),
datetime.date(2017, 10, 8)
]
next_week = canteen.get_next_week()
assert next_week == expected_range
def get_date_range():
return get_current_week() + get_next_week()
def get_date_range():
if datetime.now().weekday() > 4:
return get_next_week()
else:
return get_current_week()
def get_date_range():
today = datetime.date.today()
if today.weekday() > 4:
return get_next_week()
else:
return get_current_week()