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()