tests/perf_bench: Add perf test for yield-from execution.
Signed-off-by: Damien George <damien@micropython.org>
This commit is contained in:
parent
75da124cf8
commit
a434705700
31
tests/perf_bench/core_yield_from.py
Normal file
31
tests/perf_bench/core_yield_from.py
Normal file
@ -0,0 +1,31 @@
|
||||
# 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)
|
Loading…
Reference in New Issue
Block a user