From 0e708a1b79d644f313111ec02c9e9be500b3754f Mon Sep 17 00:00:00 2001 From: Raymond Hettinger Date: Sun, 22 Aug 2010 07:56:20 +0000 Subject: [PATCH] Issue #9214: Fix set operations on KeysView and ItemsView. --- Lib/_abcoll.py | 8 ++++++++ Lib/test/test_collections.py | 27 ++++++++++++++++++++++++++- Misc/NEWS | 3 +++ 3 files changed, 37 insertions(+), 1 deletion(-) diff --git a/Lib/_abcoll.py b/Lib/_abcoll.py index 7890e974739..d3e23c1e446 100644 --- a/Lib/_abcoll.py +++ b/Lib/_abcoll.py @@ -393,6 +393,10 @@ class MappingView(Sized): class KeysView(MappingView, Set): + @classmethod + def _from_iterable(self, it): + return set(it) + def __contains__(self, key): return key in self._mapping @@ -405,6 +409,10 @@ KeysView.register(dict_keys) class ItemsView(MappingView, Set): + @classmethod + def _from_iterable(self, it): + return set(it) + def __contains__(self, item): key, value = item try: diff --git a/Lib/test/test_collections.py b/Lib/test/test_collections.py index 85bf248a0ca..be41fcda0bc 100644 --- a/Lib/test/test_collections.py +++ b/Lib/test/test_collections.py @@ -13,7 +13,7 @@ import re from collections import Hashable, Iterable, Iterator from collections import Sized, Container, Callable from collections import Set, MutableSet -from collections import Mapping, MutableMapping +from collections import Mapping, MutableMapping, KeysView, ItemsView, UserDict from collections import Sequence, MutableSequence from collections import ByteString @@ -516,6 +516,31 @@ class TestCollectionABCs(ABCTestCase): self.validate_abstract_methods(MutableMapping, '__contains__', '__iter__', '__len__', '__getitem__', '__setitem__', '__delitem__') + def test_MutableMapping_subclass(self): + # Test issue 9214 + mymap = UserDict() + mymap['red'] = 5 + self.assert_(isinstance(mymap.keys(), Set)) + self.assert_(isinstance(mymap.keys(), KeysView)) + self.assert_(isinstance(mymap.items(), Set)) + self.assert_(isinstance(mymap.items(), ItemsView)) + + mymap = UserDict() + mymap['red'] = 5 + z = mymap.keys() | {'orange'} + self.assertEqual(type(z), set) + list(z) + mymap['blue'] = 7 # Shouldn't affect 'z' + self.assertEqual(sorted(z), ['orange', 'red']) + + mymap = UserDict() + mymap['red'] = 5 + z = mymap.items() | {('orange', 3)} + self.assertEqual(type(z), set) + list(z) + mymap['blue'] = 7 # Shouldn't affect 'z' + self.assertEqual(sorted(z), [('orange', 3), ('red', 5)]) + def test_Sequence(self): for sample in [tuple, list, bytes, str]: self.assertTrue(isinstance(sample(), Sequence)) diff --git a/Misc/NEWS b/Misc/NEWS index 9a48c5e1851..d88a4837564 100644 --- a/Misc/NEWS +++ b/Misc/NEWS @@ -95,6 +95,9 @@ C-API Library ------- +- Issue #9214: Set operations on KeysView or ItemsView in the collections + module now correctly return a set. (Patch by Eli Bendersky.) + - Issue #9617: Signals received during a low-level write operation aren't ignored by the buffered IO layer anymore.