The script now takes an optional command-line argument to specify how many
loops to run (default remains 50,000 if no argument is specified).
diff --git a/Lib/test/pystone.py b/Lib/test/pystone.py
index 931a307..168c399 100755
--- a/Lib/test/pystone.py
+++ b/Lib/test/pystone.py
@@ -57,10 +57,10 @@
TRUE = 1
FALSE = 0
-def main():
- benchtime, stones = pystones()
+def main(loops=LOOPS):
+ benchtime, stones = pystones(loops)
print "Pystone(%s) time for %d passes = %g" % \
- (__version__, LOOPS, benchtime)
+ (__version__, loops, benchtime)
print "This machine benchmarks at %g pystones/second" % stones
@@ -249,4 +249,19 @@
return FALSE
if __name__ == '__main__':
- main()
+ import sys
+ def error(msg):
+ print >>sys.stderr, msg,
+ print >>sys.stderr, "usage: %s [number_of_loops]" % sys.argv[0]
+ sys.exit(100)
+ nargs = len(sys.argv) - 1
+ if nargs > 1:
+ error("%d arguments are too many;" % nargs)
+ elif nargs == 1:
+ try: loops = int(sys.argv[1])
+ except ValueError:
+ error("Invalid argument %r;" % sys.argv[1])
+ else:
+ loops = LOOPS
+ main(loops)
+