#5228: add pickle support to functools.partial
diff --git a/Lib/test/test_functools.py b/Lib/test/test_functools.py
index 30d419d..cd645d3 100644
--- a/Lib/test/test_functools.py
+++ b/Lib/test/test_functools.py
@@ -2,6 +2,7 @@
import unittest
from test import test_support
from weakref import proxy
+import pickle
@staticmethod
def PythonPartial(func, *args, **keywords):
@@ -19,6 +20,10 @@
"""capture all positional and keyword arguments"""
return args, kw
+def signature(part):
+ """ return the signature of a partial object """
+ return (part.func, part.args, part.keywords, part.__dict__)
+
class TestPartial(unittest.TestCase):
thetype = functools.partial
@@ -140,6 +145,12 @@
join = self.thetype(''.join)
self.assertEqual(join(data), '0123456789')
+ def test_pickle(self):
+ f = self.thetype(signature, 'asdf', bar=True)
+ f.add_something_to__dict__ = True
+ f_copy = pickle.loads(pickle.dumps(f))
+ self.assertEqual(signature(f), signature(f_copy))
+
class PartialSubclass(functools.partial):
pass
@@ -147,11 +158,13 @@
thetype = PartialSubclass
-
class TestPythonPartial(TestPartial):
thetype = PythonPartial
+ # the python version isn't picklable
+ def test_pickle(self): pass
+
class TestUpdateWrapper(unittest.TestCase):
def check_wrapper(self, wrapper, wrapped,