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
|
from pypy.module.pypyjit.test_pypy_c.test_00_model import BaseTestPyPyC
class TestGenerators(BaseTestPyPyC):
def test_simple_generator1(self):
def main(n):
def f():
for i in range(10000):
i -= 1
i -= 42 # ID: subtract
yield i
def g():
for i in f(): # ID: generator
pass
g()
log = self.run(main, [500])
loop, = log.loops_by_filename(self.filepath)
assert loop.match_by_id("generator", """
cond_call(..., descr=...)
i16 = force_token()
p45 = new_with_vtable(descr=<.*>)
ifoo = arraylen_gc(p8, descr=<ArrayP .*>)
setfield_gc(p45, i29, descr=<FieldS .*>)
setarrayitem_gc(p8, 0, p45, descr=<ArrayP .>)
jump(..., descr=...)
""")
assert loop.match_by_id("subtract", """
i2 = int_sub(i1, 42)
""")
def test_simple_generator2(self):
def main(n):
def f():
for i in range(1, 10000):
i -= 1
i -= 42 # ID: subtract
yield i
def g():
for i in f(): # ID: generator
pass
g()
log = self.run(main, [500])
loop, = log.loops_by_filename(self.filepath)
assert loop.match_by_id("generator", """
cond_call(..., descr=...)
i16 = force_token()
p45 = new_with_vtable(descr=<.*>)
i47 = arraylen_gc(p8, descr=<ArrayP .>) # Should be removed by backend
setfield_gc(p45, i29, descr=<FieldS .*>)
setarrayitem_gc(p8, 0, p45, descr=<ArrayP .>)
jump(..., descr=...)
""")
assert loop.match_by_id("subtract", """
setfield_gc(p7, 38, descr=<.*last_instr .*>) # XXX bad, kill me
i2 = int_sub_ovf(i1, 42)
guard_no_overflow(descr=...)
""")
|
