1import errno 2import hotshot 3import hotshot.stats 4import sys 5import test.pystone 6 7def 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 25if __name__ == '__main__': 26 if sys.argv[1:]: 27 main(sys.argv[1]) 28 else: 29 import tempfile 30 main(tempfile.NamedTemporaryFile().name) 31