Fix repr tests for Python 2 (no string comparison)

jazzband · Apr 9, 2013 · 9c207e5 · 9c207e5
1 parent 47f6dd6
commit 9c207e5
Showing 1 changed file with 14 additions and 15 deletions.
diff --git a/model_utils/tests/tests.py b/model_utils/tests/tests.py
@@ -199,10 +199,10 @@ def test_len(self):
 
 
     def test_repr(self):
-        self.assertEquals(repr(self.STATUS),
-                          "Choices("
-                          "('DRAFT', 'DRAFT', 'DRAFT'), "
-                          "('PUBLISHED', 'PUBLISHED', 'PUBLISHED'))")
+        self.assertEquals(repr(self.STATUS), "Choices" + repr((
+            ('DRAFT', 'DRAFT', 'DRAFT'),
+            ('PUBLISHED', 'PUBLISHED', 'PUBLISHED'),
+        )))
 
 
     def test_wrong_length_tuple(self):
@@ -244,11 +244,11 @@ def test_len(self):
 
 
     def test_repr(self):
-        self.assertEquals(repr(self.STATUS),
-                          "Choices("
-                          "('DRAFT', 'DRAFT', 'is draft'), "
-                          "('PUBLISHED', 'PUBLISHED', 'is published'), "
-                          "('DELETED', 'DELETED', 'DELETED'))")
+        self.assertEquals(repr(self.STATUS), "Choices" + repr((
+            ('DRAFT', 'DRAFT', 'is draft'),
+            ('PUBLISHED', 'PUBLISHED', 'is published'),
+            ('DELETED', 'DELETED', 'DELETED'),
+        )))
 
 
 
@@ -280,12 +280,11 @@ def test_len(self):
 
 
     def test_repr(self):
-        self.assertEquals(repr(self.STATUS),
-                          "Choices("
-                          "(0, 'DRAFT', 'is draft'), "
-                          "(1, 'PUBLISHED', 'is published'), "
-                          "(2, 'DELETED', 'is deleted'))")
-
+        self.assertEquals(repr(self.STATUS), "Choices" + repr((
+            (0, 'DRAFT', 'is draft'),
+            (1, 'PUBLISHED', 'is published'),
+            (2, 'DELETED', 'is deleted'),
+        )))
 
 
 class InheritanceManagerTests(TestCase):