commit a055ced9d43630cadc3c1d5edab0884f2c5131ea
author Miss Skeleton (bot) <31488909+miss-islington@users.noreply.github.com> Sun Oct 18 08:32:56 2020 -0700
committer GitHub <noreply@github.com> Sun Oct 18 18:32:56 2020 +0300
tree 3bd0b566757d2a49bc23df33b162ae345ab1c7e2
parent 6a2aa4994e99ae8e43011a0290646fde4724a003

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

(cherry picked from commit c304c9a7efa8751b5bc7526fa95cd5f30aac2b92)

Co-authored-by: scaramallion <scaramallion@users.noreply.github.com>

Lib/datetime.py

diff --git a/Lib/datetime.py b/Lib/datetime.py
index 2294ac2..e508d99 100644
--- a/Lib/datetime.py
+++ b/Lib/datetime.py
@@ -1579,7 +1579,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 a9741d6..b37ef91 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 = [