From 3af5db0cebbe4267525af76645626b9b8570e0f2 Mon Sep 17 00:00:00 2001
From: Jerome Leclanche <jerome@leclan.ch>
Date: Tue, 8 Aug 2017 01:07:28 +0300
Subject: [PATCH] tests: Clean up and simplify test_reset_polymorphic_ctype

---
 polymorphic/tests/test_utils.py | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/polymorphic/tests/test_utils.py b/polymorphic/tests/test_utils.py
index 4c7f9a9..5a3d9b4 100644
--- a/polymorphic/tests/test_utils.py
+++ b/polymorphic/tests/test_utils.py
@@ -28,9 +28,13 @@ class UtilsTests(TestCase):
             list(Model2A.objects.all())
 
         reset_polymorphic_ctype(Model2D, Model2B, Model2D, Model2A, Model2C)
-        self.assertEqual(qrepr(Model2A.objects.order_by('pk')), (
-            '[ <Model2A: id 1, field1 (CharField)>,\n'
-            '  <Model2D: id 2, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField)>,\n'
-            '  <Model2B: id 3, field1 (CharField), field2 (CharField)>,\n'
-            '  <Model2B: id 4, field1 (CharField), field2 (CharField)> ]'
-        ))
+
+        field_reprs = [
+            "<Model2A: id 1, field1 (CharField)>",
+            "<Model2D: id 2, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField)>",
+            "<Model2B: id 3, field1 (CharField), field2 (CharField)>",
+            "<Model2B: id 4, field1 (CharField), field2 (CharField)>",
+        ]
+
+        for f, f_repr in zip(Model2A.objects.order_by("pk"), field_reprs):
+            self.assertEqual(qrepr(f), f_repr)