Refs #33476 -- Reformatted code with Black.

2 files changed, 100 insertions, 66 deletions

diff --git a/tests/many_to_many/models.py b/tests/many_to_many/models.py
index 49aa73144f..541928e94d 100644
--- a/tests/many_to_many/models.py
+++ b/tests/many_to_many/models.py
@@ -13,7 +13,7 @@ class Publication(models.Model):
     title = models.CharField(max_length=30)
 
     class Meta:
-        ordering = ('title',)
+        ordering = ("title",)
 
     def __str__(self):
         return self.title
@@ -29,21 +29,21 @@ class Tag(models.Model):
 
 class NoDeletedArticleManager(models.Manager):
     def get_queryset(self):
-        return super().get_queryset().exclude(headline='deleted')
+        return super().get_queryset().exclude(headline="deleted")
 
 
 class Article(models.Model):
     headline = models.CharField(max_length=100)
     # Assign a string as name to make sure the intermediary model is
     # correctly created. Refs #20207
-    publications = models.ManyToManyField(Publication, name='publications')
-    tags = models.ManyToManyField(Tag, related_name='tags')
-    authors = models.ManyToManyField('User', through='UserArticle')
+    publications = models.ManyToManyField(Publication, name="publications")
+    tags = models.ManyToManyField(Tag, related_name="tags")
+    authors = models.ManyToManyField("User", through="UserArticle")
 
     objects = NoDeletedArticleManager()
 
     class Meta:
-        ordering = ('headline',)
+        ordering = ("headline",)
 
     def __str__(self):
         return self.headline
@@ -57,7 +57,7 @@ class User(models.Model):
 
 
 class UserArticle(models.Model):
-    user = models.ForeignKey(User, models.CASCADE, to_field='username')
+    user = models.ForeignKey(User, models.CASCADE, to_field="username")
     article = models.ForeignKey(Article, models.CASCADE)
 
 
@@ -66,7 +66,9 @@ class AbstractArticle(models.Model):
     class Meta:
         abstract = True
 
-    publications = models.ManyToManyField(Publication, name='publications', related_name='+')
+    publications = models.ManyToManyField(
+        Publication, name="publications", related_name="+"
+    )
 
 
 class InheritedArticleA(AbstractArticle):
diff --git a/tests/many_to_many/tests.py b/tests/many_to_many/tests.py
index a20015b6d4..3d393b5711 100644
--- a/tests/many_to_many/tests.py
+++ b/tests/many_to_many/tests.py
@@ -3,50 +3,49 @@ from unittest import mock
 from django.db import transaction
 from django.test import TestCase, skipIfDBFeature, skipUnlessDBFeature
 
-from .models import (
-    Article, InheritedArticleA, InheritedArticleB, Publication, User,
-)
+from .models import Article, InheritedArticleA, InheritedArticleB, Publication, User
 
 
 class ManyToManyTests(TestCase):
-
     @classmethod
     def setUpTestData(cls):
         # Create a couple of Publications.
-        cls.p1 = Publication.objects.create(title='The Python Journal')
-        cls.p2 = Publication.objects.create(title='Science News')
-        cls.p3 = Publication.objects.create(title='Science Weekly')
-        cls.p4 = Publication.objects.create(title='Highlights for Children')
+        cls.p1 = Publication.objects.create(title="The Python Journal")
+        cls.p2 = Publication.objects.create(title="Science News")
+        cls.p3 = Publication.objects.create(title="Science Weekly")
+        cls.p4 = Publication.objects.create(title="Highlights for Children")
 
-        cls.a1 = Article.objects.create(headline='Django lets you build web apps easily')
+        cls.a1 = Article.objects.create(
+            headline="Django lets you build web apps easily"
+        )
         cls.a1.publications.add(cls.p1)
 
-        cls.a2 = Article.objects.create(headline='NASA uses Python')
+        cls.a2 = Article.objects.create(headline="NASA uses Python")
         cls.a2.publications.add(cls.p1, cls.p2, cls.p3, cls.p4)
 
-        cls.a3 = Article.objects.create(headline='NASA finds intelligent life on Earth')
+        cls.a3 = Article.objects.create(headline="NASA finds intelligent life on Earth")
         cls.a3.publications.add(cls.p2)
 
-        cls.a4 = Article.objects.create(headline='Oxygen-free diet works wonders')
+        cls.a4 = Article.objects.create(headline="Oxygen-free diet works wonders")
         cls.a4.publications.add(cls.p2)
 
     def test_add(self):
         # Create an Article.
-        a5 = Article(headline='Django lets you create web apps easily')
+        a5 = Article(headline="Django lets you create web apps easily")
         # You can't associate it with a Publication until it's been saved.
         msg = (
             '"<Article: Django lets you create web apps easily>" needs to have '
             'a value for field "id" before this many-to-many relationship can be used.'
         )
         with self.assertRaisesMessage(ValueError, msg):
-            getattr(a5, 'publications')
+            getattr(a5, "publications")
         # Save it!
         a5.save()
         # Associate the Article with a Publication.
         a5.publications.add(self.p1)
         self.assertSequenceEqual(a5.publications.all(), [self.p1])
         # Create another Article, and set it to appear in both Publications.
-        a6 = Article(headline='ESA uses Python')
+        a6 = Article(headline="ESA uses Python")
         a6.save()
         a6.publications.add(self.p1, self.p2)
         a6.publications.add(self.p3)
@@ -64,14 +63,14 @@ class ManyToManyTests(TestCase):
                 a6.publications.add(a5)
 
         # Add a Publication directly via publications.add by using keyword arguments.
-        p5 = a6.publications.create(title='Highlights for Adults')
+        p5 = a6.publications.create(title="Highlights for Adults")
         self.assertSequenceEqual(
             a6.publications.all(),
             [p5, self.p2, self.p3, self.p1],
         )
 
     def test_add_remove_set_by_pk(self):
-        a5 = Article.objects.create(headline='Django lets you create web apps easily')
+        a5 = Article.objects.create(headline="Django lets you create web apps easily")
         a5.publications.add(self.p1.pk)
         self.assertSequenceEqual(a5.publications.all(), [self.p1])
         a5.publications.set([self.p2.pk])
@@ -80,9 +79,9 @@ class ManyToManyTests(TestCase):
         self.assertSequenceEqual(a5.publications.all(), [])
 
     def test_add_remove_set_by_to_field(self):
-        user_1 = User.objects.create(username='Jean')
-        user_2 = User.objects.create(username='Joe')
-        a5 = Article.objects.create(headline='Django lets you create web apps easily')
+        user_1 = User.objects.create(username="Jean")
+        user_2 = User.objects.create(username="Joe")
+        a5 = Article.objects.create(headline="Django lets you create web apps easily")
         a5.authors.add(user_1.username)
         self.assertSequenceEqual(a5.authors.all(), [user_1])
         a5.authors.set([user_2.username])
@@ -92,13 +91,13 @@ class ManyToManyTests(TestCase):
 
     def test_add_remove_invalid_type(self):
         msg = "Field 'id' expected a number but got 'invalid'."
-        for method in ['add', 'remove']:
+        for method in ["add", "remove"]:
             with self.subTest(method), self.assertRaisesMessage(ValueError, msg):
-                getattr(self.a1.publications, method)('invalid')
+                getattr(self.a1.publications, method)("invalid")
 
     def test_reverse_add(self):
         # Adding via the 'other' end of an m2m
-        a5 = Article(headline='NASA finds intelligent life on Mars')
+        a5 = Article(headline="NASA finds intelligent life on Mars")
         a5.save()
         self.p2.article_set.add(a5)
         self.assertSequenceEqual(
@@ -108,7 +107,7 @@ class ManyToManyTests(TestCase):
         self.assertSequenceEqual(a5.publications.all(), [self.p2])
 
         # Adding via the other end using keywords
-        a6 = self.p2.article_set.create(headline='Carbon-free diet works wonders')
+        a6 = self.p2.article_set.create(headline="Carbon-free diet works wonders")
         self.assertSequenceEqual(
             self.p2.article_set.all(),
             [a6, self.a3, a5, self.a2, self.a4],
@@ -119,7 +118,7 @@ class ManyToManyTests(TestCase):
             [self.p4, self.p2, self.p3, self.p1],
         )
 
-    @skipUnlessDBFeature('supports_ignore_conflicts')
+    @skipUnlessDBFeature("supports_ignore_conflicts")
     def test_fast_add_ignore_conflicts(self):
         """
         A single query is necessary to add auto-created through instances if
@@ -129,7 +128,7 @@ class ManyToManyTests(TestCase):
         with self.assertNumQueries(1):
             self.a1.publications.add(self.p1, self.p2)
 
-    @skipIfDBFeature('supports_ignore_conflicts')
+    @skipIfDBFeature("supports_ignore_conflicts")
     def test_add_existing_different_type(self):
         # A single SELECT query is necessary to compare existing values to the
         # provided one; no INSERT should be attempted.
@@ -137,7 +136,7 @@ class ManyToManyTests(TestCase):
             self.a1.publications.add(str(self.p1.pk))
         self.assertEqual(self.a1.publications.get(), self.p1)
 
-    @skipUnlessDBFeature('supports_ignore_conflicts')
+    @skipUnlessDBFeature("supports_ignore_conflicts")
     def test_slow_add_ignore_conflicts(self):
         manager_cls = self.a1.publications.__class__
         # Simulate a race condition between the missing ids retrieval and
@@ -145,8 +144,10 @@ class ManyToManyTests(TestCase):
         missing_target_ids = {self.p1.id}
         # Disable fast-add to test the case where the slow add path is taken.
         add_plan = (True, False, False)
-        with mock.patch.object(manager_cls, '_get_missing_target_ids', return_value=missing_target_ids) as mocked:
-            with mock.patch.object(manager_cls, '_get_add_plan', return_value=add_plan):
+        with mock.patch.object(
+            manager_cls, "_get_missing_target_ids", return_value=missing_target_ids
+        ) as mocked:
+            with mock.patch.object(manager_cls, "_get_add_plan", return_value=add_plan):
                 self.a1.publications.add(self.p1)
         mocked.assert_called_once()
 
@@ -191,18 +192,29 @@ class ManyToManyTests(TestCase):
         )
         self.assertSequenceEqual(
             Article.objects.filter(publications__title__startswith="Science"),
-            [self.a3, self.a2, self.a2, self.a4]
+            [self.a3, self.a2, self.a2, self.a4],
         )
         self.assertSequenceEqual(
-            Article.objects.filter(publications__title__startswith="Science").distinct(),
+            Article.objects.filter(
+                publications__title__startswith="Science"
+            ).distinct(),
             [self.a3, self.a2, self.a4],
         )
 
         # The count() function respects distinct() as well.
-        self.assertEqual(Article.objects.filter(publications__title__startswith="Science").count(), 4)
-        self.assertEqual(Article.objects.filter(publications__title__startswith="Science").distinct().count(), 3)
+        self.assertEqual(
+            Article.objects.filter(publications__title__startswith="Science").count(), 4
+        )
+        self.assertEqual(
+            Article.objects.filter(publications__title__startswith="Science")
+            .distinct()
+            .count(),
+            3,
+        )
         self.assertSequenceEqual(
-            Article.objects.filter(publications__in=[self.p1.id, self.p2.id]).distinct(),
+            Article.objects.filter(
+                publications__in=[self.p1.id, self.p2.id]
+            ).distinct(),
             [self.a1, self.a3, self.a2, self.a4],
         )
         self.assertSequenceEqual(
@@ -225,17 +237,29 @@ class ManyToManyTests(TestCase):
         # Reverse m2m queries are supported (i.e., starting at the table that
         # doesn't have a ManyToManyField).
         python_journal = [self.p1]
-        self.assertSequenceEqual(Publication.objects.filter(id__exact=self.p1.id), python_journal)
-        self.assertSequenceEqual(Publication.objects.filter(pk=self.p1.id), python_journal)
+        self.assertSequenceEqual(
+            Publication.objects.filter(id__exact=self.p1.id), python_journal
+        )
+        self.assertSequenceEqual(
+            Publication.objects.filter(pk=self.p1.id), python_journal
+        )
         self.assertSequenceEqual(
             Publication.objects.filter(article__headline__startswith="NASA"),
             [self.p4, self.p2, self.p2, self.p3, self.p1],
         )
 
-        self.assertSequenceEqual(Publication.objects.filter(article__id__exact=self.a1.id), python_journal)
-        self.assertSequenceEqual(Publication.objects.filter(article__pk=self.a1.id), python_journal)
-        self.assertSequenceEqual(Publication.objects.filter(article=self.a1.id), python_journal)
-        self.assertSequenceEqual(Publication.objects.filter(article=self.a1), python_journal)
+        self.assertSequenceEqual(
+            Publication.objects.filter(article__id__exact=self.a1.id), python_journal
+        )
+        self.assertSequenceEqual(
+            Publication.objects.filter(article__pk=self.a1.id), python_journal
+        )
+        self.assertSequenceEqual(
+            Publication.objects.filter(article=self.a1.id), python_journal
+        )
+        self.assertSequenceEqual(
+            Publication.objects.filter(article=self.a1), python_journal
+        )
 
         self.assertSequenceEqual(
             Publication.objects.filter(article__in=[self.a1.id, self.a2.id]).distinct(),
@@ -271,7 +295,7 @@ class ManyToManyTests(TestCase):
 
     def test_bulk_delete(self):
         # Bulk delete some Publications - references to deleted publications should go
-        Publication.objects.filter(title__startswith='Science').delete()
+        Publication.objects.filter(title__startswith="Science").delete()
         self.assertSequenceEqual(
             Publication.objects.all(),
             [self.p4, self.p1],
@@ -286,7 +310,7 @@ class ManyToManyTests(TestCase):
         )
 
         # Bulk delete some articles - references to deleted objects should go
-        q = Article.objects.filter(headline__startswith='Django')
+        q = Article.objects.filter(headline__startswith="Django")
         self.assertSequenceEqual(q, [self.a1])
         q.delete()
         # After the delete, the QuerySet cache needs to be cleared,
@@ -345,14 +369,18 @@ class ManyToManyTests(TestCase):
     def test_set_existing_different_type(self):
         # Existing many-to-many relations remain the same for values provided
         # with a different type.
-        ids = set(Publication.article_set.through.objects.filter(
-            article__in=[self.a4, self.a3],
-            publication=self.p2,
-        ).values_list('id', flat=True))
+        ids = set(
+            Publication.article_set.through.objects.filter(
+                article__in=[self.a4, self.a3],
+                publication=self.p2,
+            ).values_list("id", flat=True)
+        )
         self.p2.article_set.set([str(self.a4.pk), str(self.a3.pk)])
-        new_ids = set(Publication.article_set.through.objects.filter(
-            publication=self.p2,
-        ).values_list('id', flat=True))
+        new_ids = set(
+            Publication.article_set.through.objects.filter(
+                publication=self.p2,
+            ).values_list("id", flat=True)
+        )
         self.assertEqual(ids, new_ids)
 
     def test_assign_forward(self):
@@ -407,7 +435,7 @@ class ManyToManyTests(TestCase):
         # (#19816).
         self.a1.publications.set([self.p1, self.p2])
 
-        qs = self.a1.publications.filter(title='The Python Journal')
+        qs = self.a1.publications.filter(title="The Python Journal")
         self.a1.publications.set(qs)
 
         self.assertEqual(1, self.a1.publications.count())
@@ -419,7 +447,9 @@ class ManyToManyTests(TestCase):
         # (#19816).
         self.p1.article_set.set([self.a1, self.a2])
 
-        qs = self.p1.article_set.filter(headline='Django lets you build web apps easily')
+        qs = self.p1.article_set.filter(
+            headline="Django lets you build web apps easily"
+        )
         self.p1.article_set.set(qs)
 
         self.assertEqual(1, self.p1.article_set.count())
@@ -443,25 +473,25 @@ class ManyToManyTests(TestCase):
         self.assertSequenceEqual(self.p2.article_set.all(), [self.a3])
 
     def test_clear_after_prefetch(self):
-        a4 = Article.objects.prefetch_related('publications').get(id=self.a4.id)
+        a4 = Article.objects.prefetch_related("publications").get(id=self.a4.id)
         self.assertSequenceEqual(a4.publications.all(), [self.p2])
         a4.publications.clear()
         self.assertSequenceEqual(a4.publications.all(), [])
 
     def test_remove_after_prefetch(self):
-        a4 = Article.objects.prefetch_related('publications').get(id=self.a4.id)
+        a4 = Article.objects.prefetch_related("publications").get(id=self.a4.id)
         self.assertSequenceEqual(a4.publications.all(), [self.p2])
         a4.publications.remove(self.p2)
         self.assertSequenceEqual(a4.publications.all(), [])
 
     def test_add_after_prefetch(self):
-        a4 = Article.objects.prefetch_related('publications').get(id=self.a4.id)
+        a4 = Article.objects.prefetch_related("publications").get(id=self.a4.id)
         self.assertEqual(a4.publications.count(), 1)
         a4.publications.add(self.p1)
         self.assertEqual(a4.publications.count(), 2)
 
     def test_set_after_prefetch(self):
-        a4 = Article.objects.prefetch_related('publications').get(id=self.a4.id)
+        a4 = Article.objects.prefetch_related("publications").get(id=self.a4.id)
         self.assertEqual(a4.publications.count(), 1)
         a4.publications.set([self.p2, self.p1])
         self.assertEqual(a4.publications.count(), 2)
@@ -469,7 +499,7 @@ class ManyToManyTests(TestCase):
         self.assertEqual(a4.publications.count(), 1)
 
     def test_add_then_remove_after_prefetch(self):
-        a4 = Article.objects.prefetch_related('publications').get(id=self.a4.id)
+        a4 = Article.objects.prefetch_related("publications").get(id=self.a4.id)
         self.assertEqual(a4.publications.count(), 1)
         a4.publications.add(self.p1)
         self.assertEqual(a4.publications.count(), 2)
@@ -497,7 +527,9 @@ class ManyToManyTests(TestCase):
         self.assertSequenceEqual(b.publications.all(), [self.p3])
 
     def test_custom_default_manager_exists_count(self):
-        a5 = Article.objects.create(headline='deleted')
+        a5 = Article.objects.create(headline="deleted")
         a5.publications.add(self.p2)
         self.assertEqual(self.p2.article_set.count(), self.p2.article_set.all().count())
-        self.assertEqual(self.p3.article_set.exists(), self.p3.article_set.all().exists())
+        self.assertEqual(
+            self.p3.article_set.exists(), self.p3.article_set.all().exists()
+        )