Add yappi profiling to tests/__main__.py

2011-09-24 19:09:05 +02:00 · 2011-09-24 19:09:05 +02:00 · 808b9e026a
commit 808b9e026a
parent 4438d6f42f
2 changed files with 6 additions and 1 deletions
--- a/requirements/tests.txt
+++ b/requirements/tests.txt
@ -2,3 +2,4 @@ coverage
 mock >= 0.7
 nose
 tox
+yappi
--- a/tests/main.py
+++ b/tests/main.py
@ -1,4 +1,8 @@
 import nose
+import yappi

-if __name__ == '__main__':
+try:
+    yappi.start()
    nose.main()
+finally:
+    yappi.print_stats()