diff --git a/teamanalysis/time.py b/teamanalysis/time.py index 5ab7461..c8de1bc 100644 --- a/teamanalysis/time.py +++ b/teamanalysis/time.py @@ -4,5 +4,6 @@ import datetime def get_checkpoint(n): day_of_week = n.isoweekday() new_day = n - datetime.timedelta(days=(day_of_week-1)) - checkpoint = datetime.datetime(new_day.year, new_day.month, new_day.day, 0, 0, 0, 1) - return checkpoint + start = datetime.datetime(new_day.year, new_day.month, new_day.day, 0, 0, 0, 1) + end = start + datetime.timedelta(days=7) + return start, end diff --git a/tests/test_time.py b/tests/test_time.py index 1462fe9..eece22f 100644 --- a/tests/test_time.py +++ b/tests/test_time.py @@ -8,6 +8,7 @@ import teamanalysis.time (datetime.datetime(2015, 1, 5, 2, 5, 5, 7835), datetime.datetime(2015, 1, 5, 0, 0, 0, 1)), ]) def test_checkpoint(when, expected): - checkpoint = teamanalysis.time.get_checkpoint(when) - assert checkpoint == expected + start, end = teamanalysis.time.get_checkpoint(when) + assert start == expected + assert end == start + datetime.timedelta(days=7)