Fix PY3 reference in tests

We only ever build for py3 now, the compat reference was useless.
jazzband · Aug 14, 2020 · 6be20c1 · 6be20c1
1 parent 8ae64d1
commit 6be20c1
Showing 1 changed file with 12 additions and 26 deletions.
diff --git a/polymorphic/tests/test_orm.py b/polymorphic/tests/test_orm.py
@@ -581,32 +581,18 @@ def test_extra_method(self):
             select={"topic": "tests_modelextraexternal.topic"},
             where=["tests_modelextraa.id = tests_modelextraexternal.id"],
         )
-        if compat.PY3:
-            self.assertEqual(
-                repr(objects[0]),
-                '<ModelExtraA: id 1, field1 (CharField) "A1" - Extra: topic (str) "extra1">',
-            )
-            self.assertEqual(
-                repr(objects[1]),
-                '<ModelExtraB: id 2, field1 (CharField) "B1", field2 (CharField) "B2" - Extra: topic (str) "extra2">',
-            )
-            self.assertEqual(
-                repr(objects[2]),
-                '<ModelExtraC: id 3, field1 (CharField) "C1", field2 (CharField) "C2", field3 (CharField) "C3" - Extra: topic (str) "extra3">',
-            )
-        else:
-            self.assertEqual(
-                repr(objects[0]),
-                '<ModelExtraA: id 1, field1 (CharField) "A1" - Extra: topic (unicode) "extra1">',
-            )
-            self.assertEqual(
-                repr(objects[1]),
-                '<ModelExtraB: id 2, field1 (CharField) "B1", field2 (CharField) "B2" - Extra: topic (unicode) "extra2">',
-            )
-            self.assertEqual(
-                repr(objects[2]),
-                '<ModelExtraC: id 3, field1 (CharField) "C1", field2 (CharField) "C2", field3 (CharField) "C3" - Extra: topic (unicode) "extra3">',
-            )
+        self.assertEqual(
+            repr(objects[0]),
+            '<ModelExtraA: id 1, field1 (CharField) "A1" - Extra: topic (str) "extra1">',
+        )
+        self.assertEqual(
+            repr(objects[1]),
+            '<ModelExtraB: id 2, field1 (CharField) "B1", field2 (CharField) "B2" - Extra: topic (str) "extra2">',
+        )
+        self.assertEqual(
+            repr(objects[2]),
+            '<ModelExtraC: id 3, field1 (CharField) "C1", field2 (CharField) "C2", field3 (CharField) "C3" - Extra: topic (str) "extra3">',
+        )
         self.assertEqual(len(objects), 3)
 
     def test_instance_of_filter(self):