[Python-checkins] python/dist/src/Lib/test pystone.py,1.7,1.8

[aleax at users.sourceforge.net]

Update of /cvsroot/python/python/dist/src/Lib/test
In directory sc8-pr-cvs1:/tmp/cvs-serv11491

Modified Files:
	pystone.py 
Log Message:
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).


Index: pystone.py
===================================================================
RCS file: /cvsroot/python/python/dist/src/Lib/test/pystone.py,v
retrieving revision 1.7
retrieving revision 1.8
diff -C2 -d -r1.7 -r1.8
*** pystone.py	6 Aug 2002 17:21:20 -0000	1.7
--- pystone.py	2 Jan 2004 17:11:54 -0000	1.8
***************
*** 58,65 ****
  FALSE = 0
  
! def main():
!     benchtime, stones = pystones()
      print "Pystone(%s) time for %d passes = %g" % \
!           (__version__, LOOPS, benchtime)
      print "This machine benchmarks at %g pystones/second" % stones
  
--- 58,65 ----
  FALSE = 0
  
! def main(loops=LOOPS):
!     benchtime, stones = pystones(loops)
      print "Pystone(%s) time for %d passes = %g" % \
!           (__version__, loops, benchtime)
      print "This machine benchmarks at %g pystones/second" % stones
  
***************
*** 250,252 ****
  
  if __name__ == '__main__':
!     main()
--- 250,267 ----
  
  if __name__ == '__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)
!