From 808b9e026a272a37d2d6de39be54aa80653c5480 Mon Sep 17 00:00:00 2001
From: Stein Magnus Jodal <stein.magnus@jodal.no>
Date: Sat, 24 Sep 2011 19:09:05 +0200
Subject: [PATCH] Add yappi profiling to tests/__main__.py

---
 requirements/tests.txt | 1 +
 tests/__main__.py      | 6 +++++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/requirements/tests.txt b/requirements/tests.txt
index 0bc8380f..922ef6dc 100644
--- a/requirements/tests.txt
+++ b/requirements/tests.txt
@@ -2,3 +2,4 @@ coverage
 mock >= 0.7
 nose
 tox
+yappi
diff --git a/tests/__main__.py b/tests/__main__.py
index e2bb3e72..69113580 100644
--- 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()