diff options
Diffstat (limited to 'tests/db_functions/text/test_trim.py')
| -rw-r--r-- | tests/db_functions/text/test_trim.py | 35 |
1 files changed, 19 insertions, 16 deletions
diff --git a/tests/db_functions/text/test_trim.py b/tests/db_functions/text/test_trim.py index 9834cef1c6..57a342851e 100644 --- a/tests/db_functions/text/test_trim.py +++ b/tests/db_functions/text/test_trim.py @@ -8,31 +8,34 @@ from ..models import Author class TrimTests(TestCase): def test_trim(self): - Author.objects.create(name=' John ', alias='j') - Author.objects.create(name='Rhonda', alias='r') + Author.objects.create(name=" John ", alias="j") + Author.objects.create(name="Rhonda", alias="r") authors = Author.objects.annotate( - ltrim=LTrim('name'), - rtrim=RTrim('name'), - trim=Trim('name'), + ltrim=LTrim("name"), + rtrim=RTrim("name"), + trim=Trim("name"), ) self.assertQuerysetEqual( - authors.order_by('alias'), [ - ('John ', ' John', 'John'), - ('Rhonda', 'Rhonda', 'Rhonda'), + authors.order_by("alias"), + [ + ("John ", " John", "John"), + ("Rhonda", "Rhonda", "Rhonda"), ], - lambda a: (a.ltrim, a.rtrim, a.trim) + lambda a: (a.ltrim, a.rtrim, a.trim), ) def test_trim_transform(self): - Author.objects.create(name=' John ') - Author.objects.create(name='Rhonda') + Author.objects.create(name=" John ") + Author.objects.create(name="Rhonda") tests = ( - (LTrim, 'John '), - (RTrim, ' John'), - (Trim, 'John'), + (LTrim, "John "), + (RTrim, " John"), + (Trim, "John"), ) for transform, trimmed_name in tests: with self.subTest(transform=transform): with register_lookup(CharField, transform): - authors = Author.objects.filter(**{'name__%s' % transform.lookup_name: trimmed_name}) - self.assertQuerysetEqual(authors, [' John '], lambda a: a.name) + authors = Author.objects.filter( + **{"name__%s" % transform.lookup_name: trimmed_name} + ) + self.assertQuerysetEqual(authors, [" John "], lambda a: a.name) |