Only use test import munging prior to Django 1.6's new test runner

This commit is contained in:
Tom Christie 2013-06-02 21:04:47 +01:00
parent 959ece66e5
commit a3796f430b

View File

@ -4,11 +4,13 @@ runner to pick up the tests. Yowzers.
""" """
from __future__ import unicode_literals from __future__ import unicode_literals
import os import os
import django
modules = [filename.rsplit('.', 1)[0] modules = [filename.rsplit('.', 1)[0]
for filename in os.listdir(os.path.dirname(__file__)) for filename in os.listdir(os.path.dirname(__file__))
if filename.endswith('.py') and not filename.startswith('_')] if filename.endswith('.py') and not filename.startswith('_')]
__test__ = dict() __test__ = dict()
if django.VERSION < (1, 6):
for module in modules: for module in modules:
exec("from rest_framework.tests.%s import *" % module) exec("from rest_framework.tests.%s import *" % module)