mirror of
https://github.com/python/cpython.git
synced 2025-08-04 00:48:58 +00:00
bpo-24416: Return named tuple from date.isocalendar() (GH-20113)
{date, datetime}.isocalendar() now return a private custom named tuple object IsoCalendarDate rather than a simple tuple. In order to leave IsocalendarDate as a private class and to improve what backwards compatibility is offered for pickling the result of a datetime.isocalendar() call, add a __reduce__ method to the named tuples that reduces them to plain tuples. (This is the part of this PR most likely to cause problems — if it causes major issues, switching to a strucseq or equivalent would be prudent). The pure python implementation of IsoCalendarDate uses positional-only arguments, since it is private and only constructed by position anyway; the equivalent change in the argument clinic on the C side would require us to move the forward declaration of the type above the clinic import for whatever reason, so it seems preferable to hold off on that for now. bpo-24416: https://bugs.python.org/issue24416 Original PR by Dong-hee Na with only minor alterations by Paul Ganssle. Co-authored-by: Dong-hee Na <donghee.na92@gmail.com>
This commit is contained in:
parent
aa92a7cf21
commit
1b97b9b0ad
7 changed files with 297 additions and 29 deletions
|
@ -2,6 +2,7 @@
|
|||
|
||||
See http://www.zope.org/Members/fdrake/DateTimeWiki/TestCases
|
||||
"""
|
||||
import io
|
||||
import itertools
|
||||
import bisect
|
||||
import copy
|
||||
|
@ -1355,19 +1356,43 @@ class TestDate(HarmlessMixedComparison, unittest.TestCase):
|
|||
def test_isocalendar(self):
|
||||
# Check examples from
|
||||
# http://www.phys.uu.nl/~vgent/calendar/isocalendar.htm
|
||||
for i in range(7):
|
||||
d = self.theclass(2003, 12, 22+i)
|
||||
self.assertEqual(d.isocalendar(), (2003, 52, i+1))
|
||||
d = self.theclass(2003, 12, 29) + timedelta(i)
|
||||
self.assertEqual(d.isocalendar(), (2004, 1, i+1))
|
||||
d = self.theclass(2004, 1, 5+i)
|
||||
self.assertEqual(d.isocalendar(), (2004, 2, i+1))
|
||||
d = self.theclass(2009, 12, 21+i)
|
||||
self.assertEqual(d.isocalendar(), (2009, 52, i+1))
|
||||
d = self.theclass(2009, 12, 28) + timedelta(i)
|
||||
self.assertEqual(d.isocalendar(), (2009, 53, i+1))
|
||||
d = self.theclass(2010, 1, 4+i)
|
||||
self.assertEqual(d.isocalendar(), (2010, 1, i+1))
|
||||
week_mondays = [
|
||||
((2003, 12, 22), (2003, 52, 1)),
|
||||
((2003, 12, 29), (2004, 1, 1)),
|
||||
((2004, 1, 5), (2004, 2, 1)),
|
||||
((2009, 12, 21), (2009, 52, 1)),
|
||||
((2009, 12, 28), (2009, 53, 1)),
|
||||
((2010, 1, 4), (2010, 1, 1)),
|
||||
]
|
||||
|
||||
test_cases = []
|
||||
for cal_date, iso_date in week_mondays:
|
||||
base_date = self.theclass(*cal_date)
|
||||
# Adds one test case for every day of the specified weeks
|
||||
for i in range(7):
|
||||
new_date = base_date + timedelta(i)
|
||||
new_iso = iso_date[0:2] + (iso_date[2] + i,)
|
||||
test_cases.append((new_date, new_iso))
|
||||
|
||||
for d, exp_iso in test_cases:
|
||||
with self.subTest(d=d, comparison="tuple"):
|
||||
self.assertEqual(d.isocalendar(), exp_iso)
|
||||
|
||||
# Check that the tuple contents are accessible by field name
|
||||
with self.subTest(d=d, comparison="fields"):
|
||||
t = d.isocalendar()
|
||||
self.assertEqual((t.year, t.week, t.weekday), exp_iso)
|
||||
|
||||
def test_isocalendar_pickling(self):
|
||||
"""Test that the result of datetime.isocalendar() can be pickled.
|
||||
|
||||
The result of a round trip should be a plain tuple.
|
||||
"""
|
||||
d = self.theclass(2019, 1, 1)
|
||||
p = pickle.dumps(d.isocalendar())
|
||||
res = pickle.loads(p)
|
||||
self.assertEqual(type(res), tuple)
|
||||
self.assertEqual(res, (2019, 1, 2))
|
||||
|
||||
def test_iso_long_years(self):
|
||||
# Calculate long ISO years and compare to table from
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue