commit c304c9a7efa8751b5bc7526fa95cd5f30aac2b92
author scaramallion <scaramallion@users.noreply.github.com> Mon Oct 19 01:49:48 2020 +1100
committer GitHub <noreply@github.com> Sun Oct 18 17:49:48 2020 +0300
tree 0a255fff66b458eb49c0357d36298c59e7fbf0f6
parent a0c603cb9d4dbb9909979313a88bcd1f5fde4f62

bpo-41966: Fix pickling pure datetime.time subclasses (GH-22731)

Lib/datetime.py

diff --git a/Lib/datetime.py b/Lib/datetime.py
index ea86bcb..b896b94 100644
--- a/Lib/datetime.py
+++ b/Lib/datetime.py
@@ -1548,7 +1548,7 @@
         self._tzinfo = tzinfo
 
     def __reduce_ex__(self, protocol):
-        return (time, self._getstate(protocol))
+        return (self.__class__, self._getstate(protocol))
 
     def __reduce__(self):
         return self.__reduce_ex__(2)

Lib/test/datetimetester.py

diff --git a/Lib/test/datetimetester.py b/Lib/test/datetimetester.py
index 8b61c26..775d315 100644
--- a/Lib/test/datetimetester.py
+++ b/Lib/test/datetimetester.py
@@ -1781,6 +1781,7 @@
             green = pickler.dumps(orig, proto)
             derived = unpickler.loads(green)
             self.assertEqual(orig, derived)
+            self.assertTrue(isinstance(derived, SubclassDate))
 
     def test_backdoor_resistance(self):
         # For fast unpickling, the constructor accepts a pickle byte string.
@@ -2308,6 +2309,7 @@
             green = pickler.dumps(orig, proto)
             derived = unpickler.loads(green)
             self.assertEqual(orig, derived)
+            self.assertTrue(isinstance(derived, SubclassDatetime))
 
     def test_compat_unpickle(self):
         tests = [
@@ -3357,6 +3359,7 @@
             green = pickler.dumps(orig, proto)
             derived = unpickler.loads(green)
             self.assertEqual(orig, derived)
+            self.assertTrue(isinstance(derived, SubclassTime))
 
     def test_compat_unpickle(self):
         tests = [