Skip to content

Commit 3af5db0

Browse files
committed
tests: Clean up and simplify test_reset_polymorphic_ctype
1 parent a91a04a commit 3af5db0

File tree

1 file changed

+10
-6
lines changed

1 file changed

+10
-6
lines changed

polymorphic/tests/test_utils.py

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -28,9 +28,13 @@ def test_reset_polymorphic_ctype(self):
2828
list(Model2A.objects.all())
2929

3030
reset_polymorphic_ctype(Model2D, Model2B, Model2D, Model2A, Model2C)
31-
self.assertEqual(qrepr(Model2A.objects.order_by('pk')), (
32-
'[ <Model2A: id 1, field1 (CharField)>,\n'
33-
' <Model2D: id 2, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField)>,\n'
34-
' <Model2B: id 3, field1 (CharField), field2 (CharField)>,\n'
35-
' <Model2B: id 4, field1 (CharField), field2 (CharField)> ]'
36-
))
31+
32+
field_reprs = [
33+
"<Model2A: id 1, field1 (CharField)>",
34+
"<Model2D: id 2, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField)>",
35+
"<Model2B: id 3, field1 (CharField), field2 (CharField)>",
36+
"<Model2B: id 4, field1 (CharField), field2 (CharField)>",
37+
]
38+
39+
for f, f_repr in zip(Model2A.objects.order_by("pk"), field_reprs):
40+
self.assertEqual(qrepr(f), f_repr)

0 commit comments

Comments
 (0)