def test_week_day(): assert 5 == week_day(2017, 6, 2) assert 7 == week_day(2017, 1, 1)
def _extends(self): if not self._posix_rule: return posix = self._posix_rule if not posix.dst_abbr: # std only # The future specification should match the last/default transition ttype = self._transitions[-1].ttype if not self._check_ttype(ttype, posix.std_offset, False, posix.std_abbr): raise ValueError("Posix spec does not match last transition") return if len(self._transitions) < 2: raise ValueError("Too few transitions for POSIX spec") # Extend the transitions for an additional 400 years # using the future specification # The future specification should match the last two transitions, # and those transitions should have different is_dst flags. tr0 = self._transitions[-1] tr1 = self._transitions[-2] tt0 = tr0.ttype tt1 = tr1.ttype if tt0.is_dst(): dst = tt0 std = tt1 else: dst = tt1 std = tt0 self._check_ttype(dst, posix.dst_offset, True, posix.dst_abbr) self._check_ttype(std, posix.std_offset, False, posix.std_abbr) # Add the transitions to tr1 and back to tr0 for each extra year. last_year = local_time(tr0.local, 0, 0)[0] leap_year = is_leap(last_year) jan1 = datetime(last_year, 1, 1) jan1_time = timestamp(jan1) jan1_weekday = week_day(jan1.year, jan1.month, jan1.day) % 7 if local_time(tr1.local, 0, 0)[0] != last_year: # Add a single extra transition to align to a calendar year. if tt0.is_dst(): pt1 = posix.dst_end else: pt1 = posix.dst_start tr1_offset = pt1.trans_offset(leap_year, jan1_weekday) tr = Transition(jan1_time + tr1_offset - tt0.offset, tr1.ttype, tr0) tr0 = tr tr1 = tr0 tt0 = tr0.ttype tt1 = tr1.ttype if tt0.is_dst(): pt1 = posix.dst_end pt0 = posix.dst_start else: pt1 = posix.dst_start pt0 = posix.dst_end tr = tr0 for year in range(last_year + 1, last_year + 401): jan1_time += SECS_PER_YEAR[leap_year] jan1_weekday = (jan1_weekday + DAYS_PER_YEAR[leap_year]) % 7 leap_year = not leap_year and is_leap(year) tr1_offset = pt1.trans_offset(leap_year, jan1_weekday) tr = Transition(jan1_time + tr1_offset - tt0.offset, tt1, tr) self._transitions.append(tr) tr0_offset = pt0.trans_offset(leap_year, jan1_weekday) tr = Transition(jan1_time + tr0_offset - tt1.offset, tt0, tr) self._transitions.append(tr)