From b5102e3550a589084bf33fae15bf131f47d51b0b Mon Sep 17 00:00:00 2001 From: Serhiy Storchaka Date: Tue, 29 Sep 2015 23:52:09 +0300 Subject: Issue #22958: Constructor and update method of weakref.WeakValueDictionary now accept the self and the dict keyword arguments. --- Lib/test/test_weakref.py | 25 +++++++++++++++++++++++++ Lib/weakref.py | 17 +++++++++++++++-- Misc/NEWS | 3 +++ 3 files changed, 43 insertions(+), 2 deletions(-) diff --git a/Lib/test/test_weakref.py b/Lib/test/test_weakref.py index 3e7347c..212cf34 100644 --- a/Lib/test/test_weakref.py +++ b/Lib/test/test_weakref.py @@ -1408,6 +1408,18 @@ class MappingTestCase(TestBase): dict2 = weakref.WeakValueDictionary(dict) self.assertEqual(dict[364], o) + def test_make_weak_valued_dict_misc(self): + # errors + self.assertRaises(TypeError, weakref.WeakValueDictionary.__init__) + self.assertRaises(TypeError, weakref.WeakValueDictionary, {}, {}) + self.assertRaises(TypeError, weakref.WeakValueDictionary, (), ()) + # special keyword arguments + o = Object(3) + for kw in 'self', 'dict', 'other', 'iterable': + d = weakref.WeakValueDictionary(**{kw: o}) + self.assertEqual(list(d.keys()), [kw]) + self.assertEqual(d[kw], o) + def make_weak_valued_dict(self): dict = weakref.WeakValueDictionary() objects = list(map(Object, range(self.COUNT))) @@ -1488,6 +1500,19 @@ class MappingTestCase(TestBase): def test_weak_valued_dict_update(self): self.check_update(weakref.WeakValueDictionary, {1: C(), 'a': C(), C(): C()}) + # errors + self.assertRaises(TypeError, weakref.WeakValueDictionary.update) + d = weakref.WeakValueDictionary() + self.assertRaises(TypeError, d.update, {}, {}) + self.assertRaises(TypeError, d.update, (), ()) + self.assertEqual(list(d.keys()), []) + # special keyword arguments + o = Object(3) + for kw in 'self', 'dict', 'other', 'iterable': + d = weakref.WeakValueDictionary() + d.update(**{kw: o}) + self.assertEqual(list(d.keys()), [kw]) + self.assertEqual(d[kw], o) def test_weak_keyed_dict_update(self): self.check_update(weakref.WeakKeyDictionary, diff --git a/Lib/weakref.py b/Lib/weakref.py index 12bf975..5d09497 100644 --- a/Lib/weakref.py +++ b/Lib/weakref.py @@ -98,7 +98,13 @@ class WeakValueDictionary(collections.MutableMapping): # objects are unwrapped on the way out, and we always wrap on the # way in). - def __init__(self, *args, **kw): + def __init__(*args, **kw): + if not args: + raise TypeError("descriptor '__init__' of 'WeakValueDictionary' " + "object needs an argument") + self, *args = args + if len(args) > 1: + raise TypeError('expected at most 1 arguments, got %d' % len(args)) def remove(wr, selfref=ref(self)): self = selfref() if self is not None: @@ -252,7 +258,14 @@ class WeakValueDictionary(collections.MutableMapping): else: return wr() - def update(self, dict=None, **kwargs): + def update(*args, **kwargs): + if not args: + raise TypeError("descriptor 'update' of 'WeakValueDictionary' " + "object needs an argument") + self, *args = args + if len(args) > 1: + raise TypeError('expected at most 1 arguments, got %d' % len(args)) + dict = args[0] if args else None if self._pending_removals: self._commit_removals() d = self.data diff --git a/Misc/NEWS b/Misc/NEWS index 3a9ddda..3d5dfa9 100644 --- a/Misc/NEWS +++ b/Misc/NEWS @@ -78,6 +78,9 @@ Core and Builtins Library ------- +- Issue #22958: Constructor and update method of weakref.WeakValueDictionary + now accept the self and the dict keyword arguments. + - Issue #22609: Constructor of collections.UserDict now accepts the self keyword argument. -- cgit v0.12