Backport 70140, 70141, 70143, and 70144.

Adds tests, switches from list to deque, fixes __reduce__
which was unnecessarily copying __keys.
This commit is contained in:
Raymond Hettinger 2009-03-03 22:59:25 +00:00
parent 288618e1a7
commit 131af6505a
2 changed files with 26 additions and 5 deletions

View File

@ -27,11 +27,11 @@ class OrderedDict(dict, MutableMapping):
except AttributeError:
# Note the underlying data structure for this class is likely to
# change in the future. Do not rely on it or access it directly.
self.__keys = []
self.__keys = deque()
self.update(*args, **kwds)
def clear(self):
del self.__keys[:]
self.__keys.clear()
dict.clear(self)
def __setitem__(self, key, value):
@ -58,16 +58,20 @@ class OrderedDict(dict, MutableMapping):
def __reduce__(self):
items = [[k, self[k]] for k in self]
tmp = self.__keys
del self.__keys
inst_dict = vars(self).copy()
inst_dict.pop('__keys', None)
self.__keys = tmp
if inst_dict:
return (self.__class__, (items,), inst_dict)
return self.__class__, (items,)
setdefault = MutableMapping.setdefault
update = MutableMapping.update
pop = MutableMapping.pop
def keys(self):
return self.__keys[:]
return list(self.__keys)
def values(self):
return map(self.__getitem__, self.__keys)

View File

@ -717,6 +717,23 @@ class TestOrderedDict(unittest.TestCase):
self.assertEquals(len(dup), len(od))
self.assertEquals(type(dup), type(od))
def test_yaml_linkage(self):
# Verify that __reduce__ is setup in a way that supports PyYAML's dump() feature.
# In yaml, lists are native but tuples are not.
pairs = [('c', 1), ('b', 2), ('a', 3), ('d', 4), ('e', 5), ('f', 6)]
od = OrderedDict(pairs)
# yaml.dump(od) -->
# '!!python/object/apply:__main__.OrderedDict\n- - [a, 1]\n - [b, 2]\n'
self.assert_(all(type(pair)==list for pair in od.__reduce__()[1]))
def test_reduce_not_too_fat(self):
# do not save instance dictionary if not needed
pairs = [('c', 1), ('b', 2), ('a', 3), ('d', 4), ('e', 5), ('f', 6)]
od = OrderedDict(pairs)
self.assertEqual(len(od.__reduce__()), 2)
od.x = 10
self.assertEqual(len(od.__reduce__()), 3)
def test_repr(self):
od = OrderedDict([('c', 1), ('b', 2), ('a', 3), ('d', 4), ('e', 5), ('f', 6)])
self.assertEqual(repr(od),