diff --git a/README.md b/README.md index cc4019f..4b12bea 100644 --- a/README.md +++ b/README.md @@ -7,3 +7,8 @@ from codewars_unittest import CodewarsTestRunner unittest.main(testRunner=CodewarsTestRunner()) ``` + +```python +# settings.py +TEST_RUNNER = 'codewars_unittest.django.CodewarsDjangoRunner' +``` diff --git a/codewars_unittest/django.py b/codewars_unittest/django.py new file mode 100644 index 0000000..39df824 --- /dev/null +++ b/codewars_unittest/django.py @@ -0,0 +1,8 @@ +from __future__ import absolute_import +from django.test.runner import DiscoverRunner +from .test_runner import CodewarsTestRunner + + +class CodewarsDjangoRunner(DiscoverRunner): + def run_suite(self, suite, **kwargs): + return CodewarsTestRunner().run(suite)