fix bare "py.test" runs without a directory by not defaulting to --doctest-modules which will virtually import everything

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)