1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
import pytest
from pypy.interpreter.error import OperationError
from pypy.module.cpyext.test.test_api import BaseApiTest
from pypy.module.cpyext.test.test_cpyext import AppTestCpythonExtensionBase
from pypy.module.cpyext.iterator import PyIter_Next
class TestIterator(BaseApiTest):
def test_check_iter(self, space, api):
assert api.PyIter_Check(space.iter(space.wrap("a")))
assert api.PyIter_Check(space.iter(space.newlist([])))
assert not api.PyIter_Check(space.w_type)
assert not api.PyIter_Check(space.wrap(2))
def test_getIter(self, space, api):
w_iter = api.PyObject_GetIter(space.wrap([1, 2, 3]))
assert space.unwrap(api.PyIter_Next(w_iter)) == 1
assert space.unwrap(api.PyIter_Next(w_iter)) == 2
assert space.unwrap(api.PyIter_Next(w_iter)) == 3
assert PyIter_Next(space, w_iter) is None
def test_iternext_error(self, space):
with pytest.raises(OperationError) as excinfo:
PyIter_Next(space, space.w_None)
assert excinfo.value.w_type is space.w_TypeError
class AppTestIterator(AppTestCpythonExtensionBase):
def test_noniterable_object_with_mapping_interface(self):
module = self.import_extension('foo', [
("test", "METH_NOARGS",
'''
PyObject *obj;
obj = PyObject_New(PyObject, &Foo_Type);
return obj;
'''
),
("get_dictproxy", "METH_O",
'''
return PyDictProxy_New(args);
'''
),
("check", "METH_O",
'''
return PyLong_FromLong(
PySequence_Check(args) +
PyMapping_Check(args) * 2);
''')
], prologue = '''
static PyObject *
mp_subscript(PyObject *self, PyObject *key)
{
return PyLong_FromLong(42);
}
static Py_ssize_t
mp_length(PyObject *self)
{
return 2;
}
PyMappingMethods tp_as_mapping;
static PyTypeObject Foo_Type = {
PyVarObject_HEAD_INIT(NULL, 0)
"foo.foo",
};
''', more_init = '''
Foo_Type.tp_flags = Py_TPFLAGS_DEFAULT;
Foo_Type.tp_as_mapping = &tp_as_mapping;
tp_as_mapping.mp_length = mp_length;
tp_as_mapping.mp_subscript = mp_subscript;
if (PyType_Ready(&Foo_Type) < 0) INITERROR;
''')
obj = module.test()
assert obj["hi there"] == 42
assert len(obj) == 2
assert not hasattr(obj, "__iter__")
e = raises(TypeError, iter, obj)
assert str(e.value).endswith("object is not iterable")
#
assert module.check(obj) == 2
# make sure dictionaries return false for PySequence_Check
assert module.check({'a': 1}) == 2
obj = module.get_dictproxy({'a': 10})
assert module.check(obj) == 2
def test_iterable_nonmapping_object(self):
module = self.import_extension('foo', [
("test", "METH_NOARGS",
'''
PyObject *obj;
obj = PyObject_New(PyObject, &Foo_Type);
return obj;
'''),
("check", "METH_O",
'''
return PyLong_FromLong(
PySequence_Check(args) +
PyMapping_Check(args) * 2);
''')
], prologue='''
static PyObject *
sq_item(PyObject *self, Py_ssize_t size)
{
return PyLong_FromLong(42);
}
static Py_ssize_t
sq_length(PyObject *self)
{
return 2;
}
PySequenceMethods tp_as_sequence;
static PyTypeObject Foo_Type = {
PyVarObject_HEAD_INIT(NULL, 0)
"foo.foo",
};
''', more_init='''
Foo_Type.tp_flags = Py_TPFLAGS_DEFAULT;
Foo_Type.tp_as_sequence = &tp_as_sequence;
tp_as_sequence.sq_length = sq_length;
tp_as_sequence.sq_item = sq_item;
if (PyType_Ready(&Foo_Type) < 0) INITERROR;
''')
obj = module.test()
assert obj[1] == 42
assert len(obj) == 2
assert not hasattr(obj, "__iter__")
it = iter(obj)
assert next(it) == 42
assert next(it) == 42
#
assert module.check(obj) == 1
|
