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