예제 #1
0
def test_next_business_day_weekend_returnsNextMonday(config, day):
    agent = Agent(1, [], config)

    next_monday_from_weekend = agent.shift_to_weekday(day)

    assert next_monday_from_weekend.weekday() == 0
예제 #2
0
def test_next_business_day_weekday_returnsSameDay(config, day):
    agent = Agent(1, [], config)

    same_day = agent.shift_to_weekday(day)

    assert same_day == day