Optionally import compatibility packages.

tkaemming · Sep 16, 2012 · ad67924 · ad67924
1 parent ef1f81f
commit ad67924
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/subdomains/tests/tests.py b/subdomains/tests/tests.py
@@ -7,8 +7,16 @@
 from django.test import TestCase
 from django.template import Context, Template
 
-from subdomains.compat.requestfactory import RequestFactory
-from subdomains.compat.tests import override_settings
+try:
+    from django.test.client import RequestFactory
+except ImportError:
+    from subdomains.compat.requestfactory import RequestFactory
+
+try:
+    from django.test.utils import override_settings
+except ImportError:
+    from subdomains.compat.tests import override_settings
+
 from subdomains.middleware import (SubdomainMiddleware,
     SubdomainURLRoutingMiddleware)
 from subdomains.utils import reverse, urljoin