From 1e09006c3cf85dc6605d015c55ee9debcdccb374 Mon Sep 17 00:00:00 2001 From: Serhiy Storchaka Date: Sun, 12 Jun 2016 15:08:57 +0300 Subject: [PATCH] Issue #25455: Backported tests for pickling recursive functools.partial objects. --- Lib/test/test_functools.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/Lib/test/test_functools.py b/Lib/test/test_functools.py index 69176f434e..06a9da3f41 100644 --- a/Lib/test/test_functools.py +++ b/Lib/test/test_functools.py @@ -236,6 +236,25 @@ class TestPartial(unittest.TestCase): self.assertEqual(r, ((1, 2), {})) self.assertIs(type(r[0]), tuple) + def test_recursive_pickle(self): + f = self.thetype(capture) + f.__setstate__((f, (), {}, {})) + for proto in range(pickle.HIGHEST_PROTOCOL + 1): + with self.assertRaises(RuntimeError): + pickle.dumps(f, proto) + + f = self.thetype(capture) + f.__setstate__((capture, (f,), {}, {})) + for proto in range(pickle.HIGHEST_PROTOCOL + 1): + f_copy = pickle.loads(pickle.dumps(f, proto)) + self.assertIs(f_copy.args[0], f_copy) + + f = self.thetype(capture) + f.__setstate__((capture, (), {'a': f}, {})) + for proto in range(pickle.HIGHEST_PROTOCOL + 1): + f_copy = pickle.loads(pickle.dumps(f, proto)) + self.assertIs(f_copy.keywords['a'], f_copy) + # Issue 6083: Reference counting bug def test_setstate_refcount(self): class BadSequence: @@ -270,6 +289,7 @@ class TestPythonPartial(TestPartial): test_setstate_errors = None test_setstate_subclasses = None test_setstate_refcount = None + test_recursive_pickle = None # the python version isn't deepcopyable test_deepcopy = None -- 2.50.1