SF bug 665835: filter() treatment of str and tuple inconsistent

As a side issue on this bug, it was noted that list and tuple iterators
used macros to directly access containers and would not recognize
__getitem__ overrides.  If the method is overridden, the patch returns
a generic sequence iterator which calls the __getitem__ method; otherwise,
it returns a high custom iterator with direct access to container elements.
This commit is contained in:
Raymond Hettinger 2003-04-24 16:52:47 +00:00
parent 3209410cb5
commit 9928571f3f
3 changed files with 16 additions and 0 deletions

View File

@ -256,6 +256,12 @@ def f():
yield i
vereq(list(tuple(f())), range(1000))
# Verify that __getitem__ overrides are recognized by __iter__
class T(tuple):
def __getitem__(self, key):
return str(key) + '!!!'
vereq(iter(T()).next(), '0!!!')
print '6.5.3 Lists'
# calling built-in types without argument must return empty
if list() != []: raise TestFailed,'list() does not return []'
@ -447,6 +453,12 @@ a = range(10)
a[::2] = tuple(range(5))
vereq(a, [0, 1, 1, 3, 2, 5, 3, 7, 4, 9])
# Verify that __getitem__ overrides are recognized by __iter__
class L(list):
def __getitem__(self, key):
return str(key) + '!!!'
vereq(iter(L()).next(), '0!!!')
print '6.6 Mappings == Dictionaries'
# calling built-in types without argument must return empty

View File

@ -2375,6 +2375,8 @@ list_iter(PyObject *seq)
PyErr_BadInternalCall();
return NULL;
}
if (seq->ob_type->tp_as_sequence->sq_item != list_item)
return PySeqIter_New(seq);
it = PyObject_GC_New(listiterobject, &PyListIter_Type);
if (it == NULL)
return NULL;

View File

@ -753,6 +753,8 @@ tuple_iter(PyObject *seq)
PyErr_BadInternalCall();
return NULL;
}
if (seq->ob_type->tp_as_sequence->sq_item != tupleitem)
return PySeqIter_New(seq);
it = PyObject_GC_New(tupleiterobject, &PyTupleIter_Type);
if (it == NULL)
return NULL;