mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-11-11 14:44:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

Let importlib.test.benchmark take a specific benchmark name to run.

Browse source
This commit is contained in:
Brett Cannon 2012-01-30 19:27:51 -05:00
parent 190f33cd2b
commit e3a9ae5ece
1 changed files with 18 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

21
Lib/importlib/test/benchmark.py
View file

@ -133,7 +133,7 @@ def decimal_using_bytecode(seconds, repeat):
yield result yield result
def main(import_, filename=None): def main(import_, filename=None, benchmark=None):
if filename and os.path.exists(filename): if filename and os.path.exists(filename):
with open(filename, 'r') as file: with open(filename, 'r') as file:
prev_results = json.load(file) prev_results = json.load(file)
@ -145,6 +145,14 @@ def main(import_, filename=None):
source_writing_bytecode, source_writing_bytecode,
decimal_using_bytecode, decimal_writing_bytecode, decimal_using_bytecode, decimal_writing_bytecode,
decimal_wo_bytecode,) decimal_wo_bytecode,)
if benchmark:
for b in benchmarks:
if b.__doc__ == benchmark:
benchmarks = [b]
break
else:
print('Unknown benchmark: {!r}'.format(benchmark, file=sys.stderr))
sys.exit(1)
seconds = 1 seconds = 1
seconds_plural = 's' if seconds > 1 else '' seconds_plural = 's' if seconds > 1 else ''
repeat = 3 repeat = 3
@ -189,10 +197,17 @@ if __name__ == '__main__':
parser.add_argument('-b', '--builtin', dest='builtin', action='store_true', parser.add_argument('-b', '--builtin', dest='builtin', action='store_true',
default=False, help="use the built-in __import__") default=False, help="use the built-in __import__")
parser.add_argument('-f', '--file', dest='filename', default=None, parser.add_argument('-f', '--file', dest='filename', default=None,
help='file to read/write results from/to') help='file to read/write results from/to'
'(incompatible w/ --benchmark)')
parser.add_argument('--benchmark', dest='benchmark',
help='specific benchmark to run '
'(incompatible w/ --file')
options = parser.parse_args() options = parser.parse_args()
if options.filename and options.benchmark:
print('Cannot specify a benchmark *and* read/write results')
sys.exit(1)
import_ = __import__ import_ = __import__
if not options.builtin: if not options.builtin:
import_ = importlib.__import__ import_ = importlib.__import__
main(import_, options.filename) main(import_, options.filename, options.benchmark)