From f161bc005d731107a9043e752946ec7449aa979f Mon Sep 17 00:00:00 2001 From: Eli Ribble Date: Thu, 30 Jul 2015 14:46:23 -0600 Subject: [PATCH] return start and end for a checkpoint --- teamanalysis/time.py | 5 +++-- tests/test_time.py | 5 +++-- 2 files changed, 6 insertions(+), 4 deletions(-) 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)