diff options
Diffstat (limited to 'bench')
-rw-r--r-- | bench/bench.py | 17 |
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) |