test0908
/
django
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Made prefetch_related tests use assertCountEqual() for unordered sequences.

This commit is contained in:
Jon Dufresne 2018-02-05 07:00:57 -08:00 committed by Tim Graham
parent da3df5b878
commit f0658ff818
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/prefetch_related/test_prefetch_related_objects.py
View File

@ -43,7 +43,7 @@ class PrefetchRelatedObjectsTests(TestCase):
prefetch_related_objects([book1], 'authors')
with self.assertNumQueries(0):
self.assertEqual(set(book1.authors.all()), {self.author1, self.author2, self.author3})
self.assertCountEqual(book1.authors.all(), [self.author1, self.author2, self.author3])
def test_m2m_reverse(self):
author1 = Author.objects.get(id=self.author1.id)
@ -51,7 +51,7 @@ class PrefetchRelatedObjectsTests(TestCase):
prefetch_related_objects([author1], 'books')
with self.assertNumQueries(0):
self.assertEqual(set(author1.books.all()), {self.book1, self.book2})
self.assertCountEqual(author1.books.all(), [self.book1, self.book2])
def test_foreignkey_forward(self):
authors = list(Author.objects.all())
@ -95,7 +95,7 @@ class PrefetchRelatedObjectsTests(TestCase):
prefetch_related_objects([book1], Prefetch('authors'))
with self.assertNumQueries(0):
self.assertEqual(set(book1.authors.all()), {self.author1, self.author2, self.author3})
self.assertCountEqual(book1.authors.all(), [self.author1, self.author2, self.author3])
def test_prefetch_object_to_attr(self):
book1 = Book.objects.get(id=self.book1.id)
@ -103,7 +103,7 @@ class PrefetchRelatedObjectsTests(TestCase):
prefetch_related_objects([book1], Prefetch('authors', to_attr='the_authors'))
with self.assertNumQueries(0):
self.assertEqual(set(book1.the_authors), {self.author1, self.author2, self.author3})
self.assertCountEqual(book1.the_authors, [self.author1, self.author2, self.author3])
def test_prefetch_queryset(self):
book1 = Book.objects.get(id=self.book1.id)
@ -114,4 +114,4 @@ class PrefetchRelatedObjectsTests(TestCase):
)
with self.assertNumQueries(0):
self.assertEqual(set(book1.authors.all()), {self.author1, self.author2})
self.assertCountEqual(book1.authors.all(), [self.author1, self.author2])