diff options
| -rw-r--r-- | Lib/collections.py | 8 | ||||
| -rw-r--r-- | Lib/test/test_collections.py | 6 | ||||
| -rw-r--r-- | Misc/NEWS | 3 |
3 files changed, 16 insertions, 1 deletions
diff --git a/Lib/collections.py b/Lib/collections.py index 061106b..c69f4ca 100644 --- a/Lib/collections.py +++ b/Lib/collections.py @@ -171,7 +171,6 @@ class OrderedDict(dict, MutableMapping): size += sizeof(self.__root) * n # proxy objects return size - setdefault = MutableMapping.setdefault update = MutableMapping.update pop = MutableMapping.pop keys = MutableMapping.keys @@ -179,6 +178,13 @@ class OrderedDict(dict, MutableMapping): items = MutableMapping.items __ne__ = MutableMapping.__ne__ + def setdefault(self, key, default=None): + 'OD.setdefault(k[,d]) -> OD.get(k,d), also set OD[k]=d if k not in OD' + if key in self: + return self[key] + self[key] = default + return default + @_recursive_repr() def __repr__(self): 'od.__repr__() <==> repr(od)' diff --git a/Lib/test/test_collections.py b/Lib/test/test_collections.py index 02b9dc3..32ce35b 100644 --- a/Lib/test/test_collections.py +++ b/Lib/test/test_collections.py @@ -976,6 +976,12 @@ class TestOrderedDict(unittest.TestCase): # make sure 'x' is added to the end self.assertEqual(list(od.items())[-1], ('x', 10)) + # make sure setdefault still works when __missing__ is defined + class Missing(OrderedDict): + def __missing__(self, key): + return 0 + self.assertEqual(Missing().setdefault(5, 9), 9) + def test_reinsert(self): # Given insert a, insert b, delete a, re-insert a, # verify that a is now later than b. @@ -20,6 +20,9 @@ Core and Builtins Library ------- +- Fix collections.OrderedDict.setdefault() so that it works in + subclasses that define __missing__(). + - Issue 10786: unittest.TextTestRunner default stream no longer bound at import time. `sys.stderr` now looked up at instantiation time. Fix contributed by Mark Roddy. |