Simplify test with Python2AssertMixin

incuna · Nov 28, 2014 · 0e8eabe · 0e8eabe
1 parent 560be32
commit 0e8eabe
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/user_management/models/tests/test_models.py b/user_management/models/tests/test_models.py
@@ -285,7 +285,7 @@ class InvalidUser(self.model):
         self.assertEqual(errors, expected)
 
 
-class TestCustomNameUser(TestCase):
+class TestCustomNameUser(utils.APIRequestTestCase):
     model = models.CustomNameUser
 
     def test_fields(self):
@@ -305,12 +305,7 @@ def test_fields(self):
             'avatar',
         }
 
-        try:
-            # python 3 only:
-            self.assertCountEqual(fields, expected)
-        except AttributeError:
-            # python 2 only:
-            self.assertItemsEqual(fields, expected)
+        self.assertCountEqual(fields, expected)
 
     def test_name(self):
         expected = u'Cú Chulainn'