Skip to content

Commit a82af13

Browse files
author
Adam Hrbac
committed
Rewrite test to not need the walrus operator
1 parent a439f91 commit a82af13

File tree

1 file changed

+9
-8
lines changed

1 file changed

+9
-8
lines changed

graalpython/com.oracle.graal.python.test/src/tests/test_sys_settrace.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -115,27 +115,28 @@ def gen(): # line -5
115115

116116

117117
def generator_example():
118-
for i in gen():
119-
str(i)
120-
return i
118+
x, m = gen(), {}
119+
while m.setdefault('i', next(x, False)): # this prevents CPython from tracing a StopIteration once gen() ends
120+
del m['i']
121+
return m['i']
121122

122123
generator_example.events = [((), [(0, 'generator_example', 'call', None),
123124
(1, 'generator_example', 'line', None),
125+
(2, 'generator_example', 'line', None),
124126
(-5, 'gen', 'call', None),
125127
(-4, 'gen', 'line', None),
126128
(-4, 'gen', 'return', 1),
129+
(3, 'generator_example', 'line', None),
127130
(2, 'generator_example', 'line', None),
128-
(1, 'generator_example', 'line', None),
129131
(-4, 'gen', 'call', None),
130132
(-3, 'gen', 'line', None),
131133
(-3, 'gen', 'return', 2),
134+
(3, 'generator_example', 'line', None),
132135
(2, 'generator_example', 'line', None),
133-
(1, 'generator_example', 'line', None),
134136
(-3, 'gen', 'call', None),
135137
(-3, 'gen', 'return', None),
136-
(1, 'generator_example', 'exception', StopIteration),
137-
(3, 'generator_example', 'line', None),
138-
(3, 'generator_example', 'return', 2)])]
138+
(4, 'generator_example', 'line', None),
139+
(4, 'generator_example', 'return', False)])]
139140

140141
def make_test_method(fun, name):
141142
def test_case(self):

0 commit comments

Comments
 (0)