[Python-checkins] cpython: Let importlib.test.benchmark take a specific benchmark name to run.

[brett.cannon]

http://hg.python.org/cpython/rev/2d0ff4d82b14
changeset:   74699:2d0ff4d82b14
user:        Brett Cannon <brett at python.org>
date:        Mon Jan 30 19:27:51 2012 -0500
summary:
  Let importlib.test.benchmark take a specific benchmark name to run.

files:
  Lib/importlib/test/benchmark.py |  21 ++++++++++++++++++---
  1 files changed, 18 insertions(+), 3 deletions(-)


diff --git a/Lib/importlib/test/benchmark.py b/Lib/importlib/test/benchmark.py
--- a/Lib/importlib/test/benchmark.py
+++ b/Lib/importlib/test/benchmark.py
@@ -133,7 +133,7 @@
         yield result
 
 
-def main(import_, filename=None):
+def main(import_, filename=None, benchmark=None):
     if filename and os.path.exists(filename):
         with open(filename, 'r') as file:
             prev_results = json.load(file)
@@ -145,6 +145,14 @@
                   source_writing_bytecode,
                   decimal_using_bytecode, decimal_writing_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_plural = 's' if seconds > 1 else ''
     repeat = 3
@@ -189,10 +197,17 @@
     parser.add_argument('-b', '--builtin', dest='builtin', action='store_true',
                         default=False, help="use the built-in __import__")
     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()
+    if options.filename and options.benchmark:
+        print('Cannot specify a benchmark *and* read/write results')
+        sys.exit(1)
     import_ = __import__
     if not options.builtin:
         import_ = importlib.__import__
 
-    main(import_, options.filename)
+    main(import_, options.filename, options.benchmark)

-- 
Repository URL: http://hg.python.org/cpython