You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
32 lines
639 B
32 lines
639 B
3 years ago
|
# Test a deep set of "yield from" statements.
|
||
|
|
||
|
|
||
|
def recursive_yield_from(depth, iter_):
|
||
|
if depth <= 0:
|
||
|
for i in iter_:
|
||
|
yield i
|
||
|
else:
|
||
|
yield from recursive_yield_from(depth - 1, iter_)
|
||
|
|
||
|
|
||
|
def test(n):
|
||
|
global result
|
||
|
result = 0
|
||
|
for i in recursive_yield_from(10, range(n)):
|
||
|
result += i
|
||
|
|
||
|
|
||
|
###########################################################################
|
||
|
# Benchmark interface
|
||
|
|
||
|
bm_params = {
|
||
|
(100, 10): (2000,),
|
||
|
(1000, 10): (20000,),
|
||
|
(5000, 10): (100000,),
|
||
|
}
|
||
|
|
||
|
|
||
|
def bm_setup(params):
|
||
|
(nloop,) = params
|
||
|
return lambda: test(nloop), lambda: (nloop // 100, result)
|