summaryrefslogtreecommitdiff
path: root/bench
diff options
context:
space:
mode:
Diffstat (limited to 'bench')
-rw-r--r--bench/bench.py17
1 files changed, 9 insertions, 8 deletions
diff --git a/bench/bench.py b/bench/bench.py
index 2baa92440..0a356a990 100644
--- a/bench/bench.py
+++ b/bench/bench.py
@@ -1,9 +1,10 @@
-import cProfile
-import py
-import pstats
-stats = cProfile.run('py.test.cmdline.main(["empty.py", ])', 'prof')
-p = pstats.Stats("prof")
-p.strip_dirs()
-p.sort_stats('cumulative')
-print p.print_stats(30)
+if __name__ == '__main__':
+ import cProfile
+ import py
+ import pstats
+ stats = cProfile.run('py.test.cmdline.main(["empty.py", ])', 'prof')
+ p = pstats.Stats("prof")
+ p.strip_dirs()
+ p.sort_stats('cumulative')
+ print p.print_stats(30)