def test_week_ending(): """ test calculating the beginning of a week from its end? """ pytest.debug_func() tlist = {'yesterday': daystart(time.time() - 24*3600), 'today': daystart(time.time()), 'tomorrow': daystart(time.time() + 24*3600), '2014.1104': U.epoch("2014.1104"), '2009.0401': time.mktime(time.strptime('2009.0401', '%Y.%m%d'))} for t in tlist.keys(): (start, end) = wr.week_ending(t) tm = time.localtime(tlist[t]) end_exp = time.strftime('%Y.%m%d', tm) tm = time.localtime(tlist[t] - (6*24*3600-7200)) start_exp = time.strftime('%Y.%m%d', tm) assert start_exp == start assert end_exp == end
def test_week_starting(): """ test calculating the end of a week from its start. The test for 2014.1031 spans the beginning of DST. """ pytest.debug_func() tlist = {'yesterday': time.time() - 24*3600, 'today': time.time(), 'tomorrow': time.time() + 24*3600, '2014.1028': U.epoch("2014.1028"), '2009.0401': time.mktime(time.strptime('2009.0401', '%Y.%m%d'))} for t in tlist.keys(): print("t = %s" % t) (start, end) = wr.week_starting(t) tm = time.localtime(tlist[t]) start_should_be = time.strftime('%Y.%m%d', tm) tm = time.localtime(tlist[t] + 6*24*3600 + 7200) end_should_be = time.strftime('%Y.%m%d', tm) assert start_should_be == start assert end_should_be == end