diff --git a/runtests.py b/runtests.py index ecfe17c..ee7c9e9 100644 --- a/runtests.py +++ b/runtests.py @@ -35,14 +35,19 @@ def runtests(*test_args): if hasattr(django, "setup"): django.setup() - if not test_args: - test_args = ["tests"] - parent = os.path.dirname(os.path.abspath(__file__)) sys.path.insert(0, parent) - from django.test.simple import DjangoTestSuiteRunner - failures = DjangoTestSuiteRunner( + try: + from django.test.runner import DiscoverRunner + runner_class = DiscoverRunner + test_args = ["kaleo.tests"] + except ImportError: + from django.test.simple import DjangoTestSuiteRunner + runner_class = DjangoTestSuiteRunner + test_args = ["tests"] + + failures = runner_class( verbosity=1, interactive=True, failfast=False).run_tests(test_args) sys.exit(failures)