look up __getnewargs__ and __getnewargs_ex__ on the object type (#16251)

This commit is contained in:
Benjamin Peterson 2014-02-16 13:49:16 -05:00
parent c00fa6387d
commit 2626fab4c7
3 changed files with 23 additions and 12 deletions

View file

@ -4701,6 +4701,20 @@ class PicklingTests(unittest.TestCase):
for proto in protocols:
self._check_reduce(proto, obj, listitems=list(obj))
def test_special_method_lookup(self):
protocols = range(pickle.HIGHEST_PROTOCOL + 1)
class Picky:
def __getstate__(self):
return {}
def __getattr__(self, attr):
if attr in ("__getnewargs__", "__getnewargs_ex__"):
raise AssertionError(attr)
return None
for protocol in protocols:
state = {} if protocol >= 2 else None
self._check_reduce(protocol, Picky(), state=state)
def _assert_is_copy(self, obj, objcopy, msg=None):
"""Utility method to verify if two objects are copies of each others.
"""