diff --git a/django_slowtests/testrunner.py b/django_slowtests/testrunner.py index f8789a2..25204ba 100644 --- a/django_slowtests/testrunner.py +++ b/django_slowtests/testrunner.py @@ -3,9 +3,11 @@ from unittest import TestSuite from django.test.runner import DiscoverRunner +from django.conf import settings TIMINGS = {} +NUM_SLOW_TESTS = getattr(settings, 'NUM_SLOW_TESTS', 10) def time_it(func): @@ -47,7 +49,7 @@ def teardown_test_environment(self, **kwargs): iter(TIMINGS.items()), key=operator.itemgetter(1), reverse=True - )[:10] - print("\nTen slowest tests:") + )[:NUM_SLOW_TESTS] + print("\n%s slowest tests:" % NUM_SLOW_TESTS) for func_name, timing in by_time: print(("{t:.4f}s {f}".format(f=func_name, t=timing)))