From 3793f95f98c3112ce447288a5bf9899eb9e35423 Mon Sep 17 00:00:00 2001 From: Raymond Hettinger Date: Sun, 11 Feb 2018 00:30:31 -0800 Subject: bpo-32792: Preserve mapping order in ChainMap() (GH-5586) --- Lib/collections/__init__.py | 5 ++++- Lib/test/test_collections.py | 17 +++++++++++++++++ .../Library/2018-02-08-00-47-07.bpo-32792.NtyDb4.rst | 1 + 3 files changed, 22 insertions(+), 1 deletion(-) create mode 100644 Misc/NEWS.d/next/Library/2018-02-08-00-47-07.bpo-32792.NtyDb4.rst diff --git a/Lib/collections/__init__.py b/Lib/collections/__init__.py index f0b41fd..9a753db 100644 --- a/Lib/collections/__init__.py +++ b/Lib/collections/__init__.py @@ -920,7 +920,10 @@ class ChainMap(_collections_abc.MutableMapping): return len(set().union(*self.maps)) # reuses stored hash values if possible def __iter__(self): - return iter(set().union(*self.maps)) + d = {} + for mapping in reversed(self.maps): + d.update(mapping) # reuses stored hash values if possible + return iter(d) def __contains__(self, key): return any(key in m for m in self.maps) diff --git a/Lib/test/test_collections.py b/Lib/test/test_collections.py index a55239e..2099d23 100644 --- a/Lib/test/test_collections.py +++ b/Lib/test/test_collections.py @@ -141,6 +141,23 @@ class TestChainMap(unittest.TestCase): with self.assertRaises(KeyError): d.popitem() + def test_order_preservation(self): + d = ChainMap( + OrderedDict(j=0, h=88888), + OrderedDict(), + OrderedDict(i=9999, d=4444, c=3333), + OrderedDict(f=666, b=222, g=777, c=333, h=888), + OrderedDict(), + OrderedDict(e=55, b=22), + OrderedDict(a=1, b=2, c=3, d=4, e=5), + OrderedDict(), + ) + self.assertEqual(''.join(d), 'abcdefghij') + self.assertEqual(list(d.items()), + [('a', 1), ('b', 222), ('c', 3333), ('d', 4444), + ('e', 55), ('f', 666), ('g', 777), ('h', 88888), + ('i', 9999), ('j', 0)]) + def test_dict_coercion(self): d = ChainMap(dict(a=1, b=2), dict(b=20, c=30)) self.assertEqual(dict(d), dict(a=1, b=2, c=30)) diff --git a/Misc/NEWS.d/next/Library/2018-02-08-00-47-07.bpo-32792.NtyDb4.rst b/Misc/NEWS.d/next/Library/2018-02-08-00-47-07.bpo-32792.NtyDb4.rst new file mode 100644 index 0000000..1f7df62 --- /dev/null +++ b/Misc/NEWS.d/next/Library/2018-02-08-00-47-07.bpo-32792.NtyDb4.rst @@ -0,0 +1 @@ +collections.ChainMap() preserves the order of the underlying mappings. -- cgit v0.12