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
|
from pypy.module._continuation.test.support import BaseAppTest
class AppTestGenerator(BaseAppTest):
def test_simple(self):
from _continuation import generator
#
@generator
def f(gen, n):
gen.switch(n+1)
f2(gen, n+2)
gen.switch(n+3)
#
def f2(gen, m):
gen.switch(m*2)
#
g = f(10)
res = next(g)
assert res == 11
res = next(g)
assert res == 24
res = next(g)
assert res == 13
raises(StopIteration, next, g)
def test_iterator(self):
from _continuation import generator
#
@generator
def f(gen, n):
gen.switch(n+1)
f2(gen, n+2)
gen.switch(n+3)
#
def f2(gen, m):
gen.switch(m*2)
#
res = list(f(10))
assert res == [11, 24, 13]
g = f(20)
assert iter(g) is g
def test_bound_method(self):
from _continuation import generator
#
class A(object):
def __init__(self, m):
self.m = m
#
@generator
def f(self, gen, n):
gen.switch(n - self.m)
#
a = A(10)
res = list(a.f(25))
assert res == [15]
def test_must_return_None(self):
from _continuation import generator
#
@generator
def f(gen, n):
gen.switch(n+1)
return "foo"
#
g = f(10)
res = next(g)
assert res == 11
raises(TypeError, next, g)
|
