def get_epi_data_TZ(date):
format_str = '%Y-%m-%d' # The format
dt = datetime.datetime.strptime(date, format_str).date()
epi = pm.date_to_epiweek(dt)
epi_week = epi.week
epi_day = epi.day
if epi_day >= 3: # cut off is Tuesday
epi_week = epi_week + 1
return epi.year, epi_week, epi.day
def get_epi_data(date):
# The format
format_str = '%m/%d/%y'
dt = datetime.datetime.strptime(date, format_str).date()
epi = pm.date_to_epiweek(dt)
return epi.year, epi.week, epi.day
def test_all():
epiweeks, dates = read_data()
for ew, date in zip(epiweeks, dates):
assert pm.epiweek_to_date(ew) == date
assert pm.date_to_epiweek(date) == ew
def fromReleaseDate2calendarEpiWeek(row):
calendarEW = pymmwr.date_to_epiweek(row.releaseDate)
calendarEW = "{:04d}{:02d}".format(calendarEW.year, calendarEW.week)
row['calendarEW'] = calendarEW
return row
def test_date_to_epiweek():
dt = datetime.date(2016, 11, 27)
ew = pm.Epiweek(2016, 48, 1)
assert pm.date_to_epiweek(dt) == ew