eliribble
/
teamanalysis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

return start and end for a checkpoint

This commit is contained in:
Eli Ribble 2015-07-30 14:46:23 -06:00
parent fbd519488c
commit f161bc005d
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
teamanalysis/time.py
View File

@ -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

5
tests/test_time.py
View File

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