commit 191e993365ac3206f46132dcf46236471ec54bfa
author Paul Ganssle <pganssle@users.noreply.github.com> Thu Nov 09 16:34:29 2017 -0500
committer Victor Stinner <victor.stinner@gmail.com> Thu Nov 09 13:34:29 2017 -0800
tree 56093f01ed019b5b0d6e3c6b5bc4600721b8438f
parent 72fa3014d5afa5e9ee407dee132991185da33c10

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

Lib/test/datetimetester.py

diff --git a/Lib/test/datetimetester.py b/Lib/test/datetimetester.py
index c5f91fb..5b58e99 100644
--- a/Lib/test/datetimetester.py
+++ b/Lib/test/datetimetester.py
@@ -1520,6 +1520,13 @@
         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):
@@ -2626,6 +2633,13 @@
         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):