mirror of
https://github.com/django/django.git
synced 2025-10-23 21:59:11 +00:00
Refs #33476 -- Reformatted code with Black.
This commit is contained in:
committed by
Mariusz Felisiak
parent
f68fa8b45d
commit
9c19aff7c7
@@ -6,48 +6,43 @@ from ..models import Author
|
||||
|
||||
|
||||
class SubstrTests(TestCase):
|
||||
|
||||
def test_basic(self):
|
||||
Author.objects.create(name='John Smith', alias='smithj')
|
||||
Author.objects.create(name='Rhonda')
|
||||
authors = Author.objects.annotate(name_part=Substr('name', 5, 3))
|
||||
Author.objects.create(name="John Smith", alias="smithj")
|
||||
Author.objects.create(name="Rhonda")
|
||||
authors = Author.objects.annotate(name_part=Substr("name", 5, 3))
|
||||
self.assertQuerysetEqual(
|
||||
authors.order_by('name'), [' Sm', 'da'],
|
||||
lambda a: a.name_part
|
||||
authors.order_by("name"), [" Sm", "da"], lambda a: a.name_part
|
||||
)
|
||||
authors = Author.objects.annotate(name_part=Substr('name', 2))
|
||||
authors = Author.objects.annotate(name_part=Substr("name", 2))
|
||||
self.assertQuerysetEqual(
|
||||
authors.order_by('name'), ['ohn Smith', 'honda'],
|
||||
lambda a: a.name_part
|
||||
authors.order_by("name"), ["ohn Smith", "honda"], lambda a: a.name_part
|
||||
)
|
||||
# If alias is null, set to first 5 lower characters of the name.
|
||||
Author.objects.filter(alias__isnull=True).update(
|
||||
alias=Lower(Substr('name', 1, 5)),
|
||||
alias=Lower(Substr("name", 1, 5)),
|
||||
)
|
||||
self.assertQuerysetEqual(
|
||||
authors.order_by('name'), ['smithj', 'rhond'],
|
||||
lambda a: a.alias
|
||||
authors.order_by("name"), ["smithj", "rhond"], lambda a: a.alias
|
||||
)
|
||||
|
||||
def test_start(self):
|
||||
Author.objects.create(name='John Smith', alias='smithj')
|
||||
Author.objects.create(name="John Smith", alias="smithj")
|
||||
a = Author.objects.annotate(
|
||||
name_part_1=Substr('name', 1),
|
||||
name_part_2=Substr('name', 2),
|
||||
).get(alias='smithj')
|
||||
name_part_1=Substr("name", 1),
|
||||
name_part_2=Substr("name", 2),
|
||||
).get(alias="smithj")
|
||||
|
||||
self.assertEqual(a.name_part_1[1:], a.name_part_2)
|
||||
|
||||
def test_pos_gt_zero(self):
|
||||
with self.assertRaisesMessage(ValueError, "'pos' must be greater than 0"):
|
||||
Author.objects.annotate(raises=Substr('name', 0))
|
||||
Author.objects.annotate(raises=Substr("name", 0))
|
||||
|
||||
def test_expressions(self):
|
||||
Author.objects.create(name='John Smith', alias='smithj')
|
||||
Author.objects.create(name='Rhonda')
|
||||
substr = Substr(Upper('name'), StrIndex('name', V('h')), 5)
|
||||
Author.objects.create(name="John Smith", alias="smithj")
|
||||
Author.objects.create(name="Rhonda")
|
||||
substr = Substr(Upper("name"), StrIndex("name", V("h")), 5)
|
||||
authors = Author.objects.annotate(name_part=substr)
|
||||
self.assertQuerysetEqual(
|
||||
authors.order_by('name'), ['HN SM', 'HONDA'],
|
||||
lambda a: a.name_part
|
||||
authors.order_by("name"), ["HN SM", "HONDA"], lambda a: a.name_part
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user