Lib/hotshot/stones.py

   1 import errno
   2 import hotshot
   3 import hotshot.stats
   4 import sys
   5 import test.pystone
   6
   7 def main(logfile):
   8     p = hotshot.Profile(logfile)
   9     benchtime, stones = p.runcall(test.pystone.pystones)
  10     p.close()
  11
  12     print "Pystone(%s) time for %d passes = %g" % \
  13           (test.pystone.__version__, test.pystone.LOOPS, benchtime)
  14     print "This machine benchmarks at %g pystones/second" % stones
  15
  16     stats = hotshot.stats.load(logfile)
  17     stats.strip_dirs()
  18     stats.sort_stats('time', 'calls')
  19     try:
  20         stats.print_stats(20)
  21     except IOError, e:
  22         if e.errno != errno.EPIPE:
  23             raise
  24
  25 if __name__ == '__main__':
  26     if sys.argv[1:]:
  27         main(sys.argv[1])
  28     else:
  29         import tempfile
  30         main(tempfile.NamedTemporaryFile().name)