More version craziness

revsys · May 23, 2015 · 4ab0984 · 4ab0984
1 parent 1ec6290
commit 4ab0984
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/test_plus/test.py b/test_plus/test.py
@@ -3,9 +3,9 @@
 from django.core.urlresolvers import reverse
 from django.db import connections, DEFAULT_DB_ALIAS
 from django.test import TestCase
+from distutils.version import LooseVersion
 
-
-if django.VERSION[0:2] >= 1.6:
+if LooseVersion(django.get_version()) >= LooseVersion('1.6'):
     from django.test.utils import CaptureQueriesContext
     CAPTURE = True
 else:
@@ -116,7 +116,7 @@ def make_user(self, username):
             test_user.save()
             return test_user
         else:
-            if django.VERSION[0:2] >= 1.6:
+            if LooseVersion(django.get_version()) >= LooseVersion('1.6'):
                 from django.contrib.auth import get_user_model
                 User = get_user_model()
             else:

diff --git a/test_project/test_app/tests.py b/test_project/test_app/tests.py
@@ -1,9 +1,10 @@
 import django
 import factory
+from distutils.version import LooseVersion
 
 from test_plus.test import TestCase
 
-if django.VERSION[0:2] >= 1.6:
+if LooseVersion(django.get_version()) >= LooseVersion('1.6'):
     from django.contrib.auth import get_user_model
     User = get_user_model()
 else: