mirror of
https://github.com/python/cpython.git
synced 2025-09-26 18:29:57 +00:00
Issue #5647: MutableSet.__iand__() no longer mutates self during iteration.
This commit is contained in:
parent
3e7a65f565
commit
38f57a1c75
2 changed files with 28 additions and 4 deletions
|
@ -286,9 +286,8 @@ class MutableSet(Set):
|
||||||
self.add(value)
|
self.add(value)
|
||||||
return self
|
return self
|
||||||
|
|
||||||
def __iand__(self, c):
|
def __iand__(self, it):
|
||||||
for value in self:
|
for value in (self - it):
|
||||||
if value not in c:
|
|
||||||
self.discard(value)
|
self.discard(value)
|
||||||
return self
|
return self
|
||||||
|
|
||||||
|
|
|
@ -311,6 +311,25 @@ class TestOneTrickPonyABCs(ABCTestCase):
|
||||||
B.register(C)
|
B.register(C)
|
||||||
self.failUnless(issubclass(C, B))
|
self.failUnless(issubclass(C, B))
|
||||||
|
|
||||||
|
class WithSet(MutableSet):
|
||||||
|
|
||||||
|
def __init__(self, it=()):
|
||||||
|
self.data = set(it)
|
||||||
|
|
||||||
|
def __len__(self):
|
||||||
|
return len(self.data)
|
||||||
|
|
||||||
|
def __iter__(self):
|
||||||
|
return iter(self.data)
|
||||||
|
|
||||||
|
def __contains__(self, item):
|
||||||
|
return item in self.data
|
||||||
|
|
||||||
|
def add(self, item):
|
||||||
|
self.data.add(item)
|
||||||
|
|
||||||
|
def discard(self, item):
|
||||||
|
self.data.discard(item)
|
||||||
|
|
||||||
class TestCollectionABCs(ABCTestCase):
|
class TestCollectionABCs(ABCTestCase):
|
||||||
|
|
||||||
|
@ -347,6 +366,12 @@ class TestCollectionABCs(ABCTestCase):
|
||||||
self.validate_abstract_methods(MutableSet, '__contains__', '__iter__', '__len__',
|
self.validate_abstract_methods(MutableSet, '__contains__', '__iter__', '__len__',
|
||||||
'add', 'discard')
|
'add', 'discard')
|
||||||
|
|
||||||
|
def test_issue_5647(self):
|
||||||
|
# MutableSet.__iand__ mutated the set during iteration
|
||||||
|
s = WithSet('abcd')
|
||||||
|
s &= WithSet('cdef') # This used to fail
|
||||||
|
self.assertEqual(set(s), set('cd'))
|
||||||
|
|
||||||
def test_issue_4920(self):
|
def test_issue_4920(self):
|
||||||
# MutableSet.pop() method did not work
|
# MutableSet.pop() method did not work
|
||||||
class MySet(collections.MutableSet):
|
class MySet(collections.MutableSet):
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue