bpo-31222: Make (datetime|date|time).replace return subclass type in Pure Python (GH-4176) (#4356)

(cherry picked from commit 191e993365)
This commit is contained in:
Miss Islington (bot) 2017-11-09 16:52:05 -08:00 committed by Victor Stinner
parent 596286f8f3
commit b9a40aca29
2 changed files with 17 additions and 3 deletions

View File

@ -827,7 +827,7 @@ class date:
month = self._month
if day is None:
day = self._day
return date(year, month, day)
return type(self)(year, month, day)
# Comparisons of date objects with other.
@ -1315,7 +1315,7 @@ class time:
tzinfo = self.tzinfo
if fold is None:
fold = self._fold
return time(hour, minute, second, microsecond, tzinfo, fold=fold)
return type(self)(hour, minute, second, microsecond, tzinfo, fold=fold)
# Pickle support.
@ -1596,7 +1596,7 @@ class datetime(date):
tzinfo = self.tzinfo
if fold is None:
fold = self.fold
return datetime(year, month, day, hour, minute, second,
return type(self)(year, month, day, hour, minute, second,
microsecond, tzinfo, fold=fold)
def _local_timezone(self):

View File

@ -1500,6 +1500,13 @@ class TestDate(HarmlessMixedComparison, unittest.TestCase):
base = cls(2000, 2, 29)
self.assertRaises(ValueError, base.replace, year=2001)
def test_subclass_replace(self):
class DateSubclass(self.theclass):
pass
dt = DateSubclass(2012, 1, 1)
self.assertIs(type(dt.replace(year=2013)), DateSubclass)
def test_subclass_date(self):
class C(self.theclass):
@ -2599,6 +2606,13 @@ class TestTime(HarmlessMixedComparison, unittest.TestCase):
self.assertRaises(ValueError, base.replace, second=100)
self.assertRaises(ValueError, base.replace, microsecond=1000000)
def test_subclass_replace(self):
class TimeSubclass(self.theclass):
pass
ctime = TimeSubclass(12, 30)
self.assertIs(type(ctime.replace(hour=10)), TimeSubclass)
def test_subclass_time(self):
class C(self.theclass):