Move testing code into "if __name__ == '__main__'" so it's not run on import.

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)