diff options
author | Ka-Ping Yee <ping@zesty.ca> | 2003-03-28 16:28:48 (GMT) |
---|---|---|
committer | Ka-Ping Yee <ping@zesty.ca> | 2003-03-28 16:28:48 (GMT) |
commit | e9638ccadbd9fc16b5d68e2e51fd0cae1a81e978 (patch) | |
tree | d6ee09664ce78511968e76a7ca3665cced202069 /Lib/hotshot/stones.py | |
parent | 5b75c38cc98762d59593ea1ce1d4d02946d499b4 (diff) | |
download | cpython-e9638ccadbd9fc16b5d68e2e51fd0cae1a81e978.zip cpython-e9638ccadbd9fc16b5d68e2e51fd0cae1a81e978.tar.gz cpython-e9638ccadbd9fc16b5d68e2e51fd0cae1a81e978.tar.bz2 |
Move testing code into "if __name__ == '__main__'" so it's not run on import.
Diffstat (limited to 'Lib/hotshot/stones.py')
-rw-r--r-- | Lib/hotshot/stones.py | 42 |
1 files changed, 21 insertions, 21 deletions
diff --git a/Lib/hotshot/stones.py b/Lib/hotshot/stones.py index e171fbc..cd4c51d 100644 --- a/Lib/hotshot/stones.py +++ b/Lib/hotshot/stones.py @@ -5,27 +5,27 @@ import os import sys import test.pystone +def main(logfile): + p = hotshot.Profile(logfile) + benchtime, stones = p.runcall(test.pystone.pystones) + p.close() -if sys.argv[1:]: - logfile = sys.argv[1] -else: - import tempfile - logf = tempfile.NamedTemporaryFile() - logfile = logf.name + print "Pystone(%s) time for %d passes = %g" % \ + (test.pystone.__version__, test.pystone.LOOPS, benchtime) + print "This machine benchmarks at %g pystones/second" % stones -p = hotshot.Profile(logfile) -benchtime, stones = p.runcall(test.pystone.pystones) -p.close() + stats = hotshot.stats.load(logfile) + stats.strip_dirs() + stats.sort_stats('time', 'calls') + try: + stats.print_stats(20) + except IOError, e: + if e.errno != errno.EPIPE: + raise -print "Pystone(%s) time for %d passes = %g" % \ - (test.pystone.__version__, test.pystone.LOOPS, benchtime) -print "This machine benchmarks at %g pystones/second" % stones - -stats = hotshot.stats.load(logfile) -stats.strip_dirs() -stats.sort_stats('time', 'calls') -try: - stats.print_stats(20) -except IOError, e: - if e.errno != errno.EPIPE: - raise +if __name__ == '__main__': + if sys.argv[1:]: + main(sys.argv[1]) + else: + import tempfile + main(tempfile.NamedTemporaryFile().name) |