Add way to get clear checkpoints for starting new time periods
This commit is contained in:
parent
c33e9b3b20
commit
fbd519488c
|
@ -0,0 +1,3 @@
|
|||
[pytest]
|
||||
norecursedirs = ve
|
||||
addopts=--tb=short
|
|
@ -0,0 +1,8 @@
|
|||
import calendar
|
||||
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
|
|
@ -0,0 +1,13 @@
|
|||
import datetime
|
||||
import pytest
|
||||
import teamanalysis.time
|
||||
|
||||
@pytest.mark.parametrize('when, expected', [
|
||||
(datetime.datetime(2015, 7, 30, 20, 35, 35, 784435), datetime.datetime(2015, 7, 27, 0, 0, 0, 1)),
|
||||
(datetime.datetime(2015, 1, 3, 2, 5, 5, 7835), datetime.datetime(2014, 12, 29, 0, 0, 0, 1)),
|
||||
(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
|
||||
|
Loading…
Reference in New Issue