summaryrefslogtreecommitdiffstats
path: root/Lib/hotshot/stones.py
diff options
context:
space:
mode:
authorKa-Ping Yee <ping@zesty.ca>2003-03-28 16:28:48 (GMT)
committerKa-Ping Yee <ping@zesty.ca>2003-03-28 16:28:48 (GMT)
commite9638ccadbd9fc16b5d68e2e51fd0cae1a81e978 (patch)
treed6ee09664ce78511968e76a7ca3665cced202069 /Lib/hotshot/stones.py
parent5b75c38cc98762d59593ea1ce1d4d02946d499b4 (diff)
downloadcpython-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.py42
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)