Issue #22958: Constructor and update method of weakref.WeakValueDictionary
now accept the self and the dict keyword arguments.
This commit is contained in:
commit
b27232949d
|
@ -1421,6 +1421,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)))
|
||||
|
@ -1501,6 +1513,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,
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue