diff options
author | django-bot <ops@djangoproject.com> | 2022-02-03 20:24:19 +0100 |
---|---|---|
committer | Mariusz Felisiak <felisiak.mariusz@gmail.com> | 2022-02-07 20:37:05 +0100 |
commit | 9c19aff7c7561e3a82978a272ecdaad40dda5c00 (patch) | |
tree | f0506b668a013d0063e5fba3dbf4863b466713ba /tests/or_lookups | |
parent | f68fa8b45dfac545cfc4111d4e52804c86db68d3 (diff) | |
download | django-9c19aff7c7561e3a82978a272ecdaad40dda5c00.tar.gz |
Refs #33476 -- Reformatted code with Black.
Diffstat (limited to 'tests/or_lookups')
-rw-r--r-- | tests/or_lookups/models.py | 2 | ||||
-rw-r--r-- | tests/or_lookups/tests.py | 198 |
2 files changed, 91 insertions, 109 deletions
diff --git a/tests/or_lookups/models.py b/tests/or_lookups/models.py index bd1840338f..76090da9f3 100644 --- a/tests/or_lookups/models.py +++ b/tests/or_lookups/models.py @@ -16,7 +16,7 @@ class Article(models.Model): pub_date = models.DateTimeField() class Meta: - ordering = ('pub_date',) + ordering = ("pub_date",) def __str__(self): return self.headline diff --git a/tests/or_lookups/tests.py b/tests/or_lookups/tests.py index f2d2ec2fad..1a7264015c 100644 --- a/tests/or_lookups/tests.py +++ b/tests/or_lookups/tests.py @@ -8,57 +8,48 @@ from .models import Article class OrLookupsTests(TestCase): - @classmethod def setUpTestData(cls): cls.a1 = Article.objects.create( - headline='Hello', pub_date=datetime(2005, 11, 27) + headline="Hello", pub_date=datetime(2005, 11, 27) ).pk cls.a2 = Article.objects.create( - headline='Goodbye', pub_date=datetime(2005, 11, 28) + headline="Goodbye", pub_date=datetime(2005, 11, 28) ).pk cls.a3 = Article.objects.create( - headline='Hello and goodbye', pub_date=datetime(2005, 11, 29) + headline="Hello and goodbye", pub_date=datetime(2005, 11, 29) ).pk def test_filter_or(self): self.assertQuerysetEqual( ( - Article.objects.filter(headline__startswith='Hello') | - Article.objects.filter(headline__startswith='Goodbye') - ), [ - 'Hello', - 'Goodbye', - 'Hello and goodbye' - ], - attrgetter("headline") + Article.objects.filter(headline__startswith="Hello") + | Article.objects.filter(headline__startswith="Goodbye") + ), + ["Hello", "Goodbye", "Hello and goodbye"], + attrgetter("headline"), ) self.assertQuerysetEqual( - Article.objects.filter(headline__contains='Hello') | Article.objects.filter(headline__contains='bye'), [ - 'Hello', - 'Goodbye', - 'Hello and goodbye' - ], - attrgetter("headline") + Article.objects.filter(headline__contains="Hello") + | Article.objects.filter(headline__contains="bye"), + ["Hello", "Goodbye", "Hello and goodbye"], + attrgetter("headline"), ) self.assertQuerysetEqual( - Article.objects.filter(headline__iexact='Hello') | Article.objects.filter(headline__contains='ood'), [ - 'Hello', - 'Goodbye', - 'Hello and goodbye' - ], - attrgetter("headline") + Article.objects.filter(headline__iexact="Hello") + | Article.objects.filter(headline__contains="ood"), + ["Hello", "Goodbye", "Hello and goodbye"], + attrgetter("headline"), ) self.assertQuerysetEqual( - Article.objects.filter(Q(headline__startswith='Hello') | Q(headline__startswith='Goodbye')), [ - 'Hello', - 'Goodbye', - 'Hello and goodbye' - ], - attrgetter("headline") + Article.objects.filter( + Q(headline__startswith="Hello") | Q(headline__startswith="Goodbye") + ), + ["Hello", "Goodbye", "Hello and goodbye"], + attrgetter("headline"), ) def test_stages(self): @@ -66,59 +57,46 @@ class OrLookupsTests(TestCase): # especially useful if building the query in stages: articles = Article.objects.all() self.assertQuerysetEqual( - articles.filter(headline__startswith='Hello') & articles.filter(headline__startswith='Goodbye'), - [] + articles.filter(headline__startswith="Hello") + & articles.filter(headline__startswith="Goodbye"), + [], ) self.assertQuerysetEqual( - articles.filter(headline__startswith='Hello') & articles.filter(headline__contains='bye'), [ - 'Hello and goodbye' - ], - attrgetter("headline") + articles.filter(headline__startswith="Hello") + & articles.filter(headline__contains="bye"), + ["Hello and goodbye"], + attrgetter("headline"), ) def test_pk_q(self): self.assertQuerysetEqual( - Article.objects.filter(Q(pk=self.a1) | Q(pk=self.a2)), [ - 'Hello', - 'Goodbye' - ], - attrgetter("headline") + Article.objects.filter(Q(pk=self.a1) | Q(pk=self.a2)), + ["Hello", "Goodbye"], + attrgetter("headline"), ) self.assertQuerysetEqual( - Article.objects.filter(Q(pk=self.a1) | Q(pk=self.a2) | Q(pk=self.a3)), [ - 'Hello', - 'Goodbye', - 'Hello and goodbye' - ], + Article.objects.filter(Q(pk=self.a1) | Q(pk=self.a2) | Q(pk=self.a3)), + ["Hello", "Goodbye", "Hello and goodbye"], attrgetter("headline"), ) def test_pk_in(self): self.assertQuerysetEqual( - Article.objects.filter(pk__in=[self.a1, self.a2, self.a3]), [ - 'Hello', - 'Goodbye', - 'Hello and goodbye' - ], + Article.objects.filter(pk__in=[self.a1, self.a2, self.a3]), + ["Hello", "Goodbye", "Hello and goodbye"], attrgetter("headline"), ) self.assertQuerysetEqual( - Article.objects.filter(pk__in=(self.a1, self.a2, self.a3)), [ - 'Hello', - 'Goodbye', - 'Hello and goodbye' - ], + Article.objects.filter(pk__in=(self.a1, self.a2, self.a3)), + ["Hello", "Goodbye", "Hello and goodbye"], attrgetter("headline"), ) self.assertQuerysetEqual( - Article.objects.filter(pk__in=[self.a1, self.a2, self.a3, 40000]), [ - 'Hello', - 'Goodbye', - 'Hello and goodbye' - ], + Article.objects.filter(pk__in=[self.a1, self.a2, self.a3, 40000]), + ["Hello", "Goodbye", "Hello and goodbye"], attrgetter("headline"), ) @@ -131,25 +109,21 @@ class OrLookupsTests(TestCase): def test_q_negated(self): # Q objects can be negated self.assertQuerysetEqual( - Article.objects.filter(Q(pk=self.a1) | ~Q(pk=self.a2)), [ - 'Hello', - 'Hello and goodbye' - ], - attrgetter("headline") + Article.objects.filter(Q(pk=self.a1) | ~Q(pk=self.a2)), + ["Hello", "Hello and goodbye"], + attrgetter("headline"), ) self.assertQuerysetEqual( - Article.objects.filter(~Q(pk=self.a1) & ~Q(pk=self.a2)), [ - 'Hello and goodbye' - ], + Article.objects.filter(~Q(pk=self.a1) & ~Q(pk=self.a2)), + ["Hello and goodbye"], attrgetter("headline"), ) # This allows for more complex queries than filter() and exclude() # alone would allow self.assertQuerysetEqual( - Article.objects.filter(Q(pk=self.a1) & (~Q(pk=self.a2) | Q(pk=self.a3))), [ - 'Hello' - ], + Article.objects.filter(Q(pk=self.a1) & (~Q(pk=self.a2) | Q(pk=self.a3))), + ["Hello"], attrgetter("headline"), ) @@ -158,83 +132,91 @@ class OrLookupsTests(TestCase): # 'limit_choices_to' which normally take a single dictionary of lookup # arguments but need to support arbitrary queries via Q objects too. self.assertQuerysetEqual( - Article.objects.complex_filter({'pk': self.a1}), [ - 'Hello' - ], + Article.objects.complex_filter({"pk": self.a1}), + ["Hello"], attrgetter("headline"), ) self.assertQuerysetEqual( - Article.objects.complex_filter(Q(pk=self.a1) | Q(pk=self.a2)), [ - 'Hello', - 'Goodbye' - ], + Article.objects.complex_filter(Q(pk=self.a1) | Q(pk=self.a2)), + ["Hello", "Goodbye"], attrgetter("headline"), ) def test_empty_in(self): # Passing "in" an empty list returns no results ... - self.assertQuerysetEqual( - Article.objects.filter(pk__in=[]), - [] - ) + self.assertQuerysetEqual(Article.objects.filter(pk__in=[]), []) # ... but can return results if we OR it with another query. self.assertQuerysetEqual( - Article.objects.filter(Q(pk__in=[]) | Q(headline__icontains='goodbye')), [ - 'Goodbye', - 'Hello and goodbye' - ], + Article.objects.filter(Q(pk__in=[]) | Q(headline__icontains="goodbye")), + ["Goodbye", "Hello and goodbye"], attrgetter("headline"), ) def test_q_and(self): # Q arg objects are ANDed self.assertQuerysetEqual( - Article.objects.filter(Q(headline__startswith='Hello'), Q(headline__contains='bye')), [ - 'Hello and goodbye' - ], - attrgetter("headline") + Article.objects.filter( + Q(headline__startswith="Hello"), Q(headline__contains="bye") + ), + ["Hello and goodbye"], + attrgetter("headline"), ) # Q arg AND order is irrelevant self.assertQuerysetEqual( - Article.objects.filter(Q(headline__contains='bye'), headline__startswith='Hello'), [ - 'Hello and goodbye' - ], + Article.objects.filter( + Q(headline__contains="bye"), headline__startswith="Hello" + ), + ["Hello and goodbye"], attrgetter("headline"), ) self.assertQuerysetEqual( - Article.objects.filter(Q(headline__startswith='Hello') & Q(headline__startswith='Goodbye')), - [] + Article.objects.filter( + Q(headline__startswith="Hello") & Q(headline__startswith="Goodbye") + ), + [], ) def test_q_exclude(self): self.assertQuerysetEqual( - Article.objects.exclude(Q(headline__startswith='Hello')), [ - 'Goodbye' - ], - attrgetter("headline") + Article.objects.exclude(Q(headline__startswith="Hello")), + ["Goodbye"], + attrgetter("headline"), ) def test_other_arg_queries(self): # Try some arg queries with operations other than filter. self.assertEqual( - Article.objects.get(Q(headline__startswith='Hello'), Q(headline__contains='bye')).headline, - 'Hello and goodbye' + Article.objects.get( + Q(headline__startswith="Hello"), Q(headline__contains="bye") + ).headline, + "Hello and goodbye", ) self.assertEqual( - Article.objects.filter(Q(headline__startswith='Hello') | Q(headline__contains='bye')).count(), - 3 + Article.objects.filter( + Q(headline__startswith="Hello") | Q(headline__contains="bye") + ).count(), + 3, ) self.assertSequenceEqual( - Article.objects.filter(Q(headline__startswith='Hello'), Q(headline__contains='bye')).values(), [ - {"headline": "Hello and goodbye", "id": self.a3, "pub_date": datetime(2005, 11, 29)}, + Article.objects.filter( + Q(headline__startswith="Hello"), Q(headline__contains="bye") + ).values(), + [ + { + "headline": "Hello and goodbye", + "id": self.a3, + "pub_date": datetime(2005, 11, 29), + }, ], ) self.assertEqual( - Article.objects.filter(Q(headline__startswith='Hello')).in_bulk([self.a1, self.a2]), - {self.a1: Article.objects.get(pk=self.a1)} + Article.objects.filter(Q(headline__startswith="Hello")).in_bulk( + [self.a1, self.a2] + ), + {self.a1: Article.objects.get(pk=self.a1)}, ) |