diff options
| author | django-bot <ops@djangoproject.com> | 2022-02-08 12:09:55 +0100 |
|---|---|---|
| committer | Mariusz Felisiak <felisiak.mariusz@gmail.com> | 2022-02-08 12:15:38 +0100 |
| commit | 6a682b38e75d4c975b4c4493565a59f1bc14397c (patch) | |
| tree | 0bd9cda550bea26238656d9f120d769e8b41bb9e /tests/db_functions/text/test_replace.py | |
| parent | e73ce08888e6f34d3f050377cfd2fbb733be94a9 (diff) | |
[4.0.x] Refs #33476 -- Reformatted code with Black.
Backport of 9c19aff7c7561e3a82978a272ecdaad40dda5c00 from main.
Diffstat (limited to 'tests/db_functions/text/test_replace.py')
| -rw-r--r-- | tests/db_functions/text/test_replace.py | 86 |
1 files changed, 57 insertions, 29 deletions
diff --git a/tests/db_functions/text/test_replace.py b/tests/db_functions/text/test_replace.py index ae87781b8c..28e39faf09 100644 --- a/tests/db_functions/text/test_replace.py +++ b/tests/db_functions/text/test_replace.py @@ -6,50 +6,78 @@ from ..models import Author class ReplaceTests(TestCase): - @classmethod def setUpTestData(cls): - Author.objects.create(name='George R. R. Martin') - Author.objects.create(name='J. R. R. Tolkien') + Author.objects.create(name="George R. R. Martin") + Author.objects.create(name="J. R. R. Tolkien") def test_replace_with_empty_string(self): qs = Author.objects.annotate( - without_middlename=Replace(F('name'), Value('R. R. '), Value('')), + without_middlename=Replace(F("name"), Value("R. R. "), Value("")), + ) + self.assertQuerysetEqual( + qs, + [ + ("George R. R. Martin", "George Martin"), + ("J. R. R. Tolkien", "J. Tolkien"), + ], + transform=lambda x: (x.name, x.without_middlename), + ordered=False, ) - self.assertQuerysetEqual(qs, [ - ('George R. R. Martin', 'George Martin'), - ('J. R. R. Tolkien', 'J. Tolkien'), - ], transform=lambda x: (x.name, x.without_middlename), ordered=False) def test_case_sensitive(self): - qs = Author.objects.annotate(same_name=Replace(F('name'), Value('r. r.'), Value(''))) - self.assertQuerysetEqual(qs, [ - ('George R. R. Martin', 'George R. R. Martin'), - ('J. R. R. Tolkien', 'J. R. R. Tolkien'), - ], transform=lambda x: (x.name, x.same_name), ordered=False) + qs = Author.objects.annotate( + same_name=Replace(F("name"), Value("r. r."), Value("")) + ) + self.assertQuerysetEqual( + qs, + [ + ("George R. R. Martin", "George R. R. Martin"), + ("J. R. R. Tolkien", "J. R. R. Tolkien"), + ], + transform=lambda x: (x.name, x.same_name), + ordered=False, + ) def test_replace_expression(self): - qs = Author.objects.annotate(same_name=Replace( - Concat(Value('Author: '), F('name')), Value('Author: '), Value('')), + qs = Author.objects.annotate( + same_name=Replace( + Concat(Value("Author: "), F("name")), Value("Author: "), Value("") + ), + ) + self.assertQuerysetEqual( + qs, + [ + ("George R. R. Martin", "George R. R. Martin"), + ("J. R. R. Tolkien", "J. R. R. Tolkien"), + ], + transform=lambda x: (x.name, x.same_name), + ordered=False, ) - self.assertQuerysetEqual(qs, [ - ('George R. R. Martin', 'George R. R. Martin'), - ('J. R. R. Tolkien', 'J. R. R. Tolkien'), - ], transform=lambda x: (x.name, x.same_name), ordered=False) def test_update(self): Author.objects.update( - name=Replace(F('name'), Value('R. R. '), Value('')), + name=Replace(F("name"), Value("R. R. "), Value("")), + ) + self.assertQuerysetEqual( + Author.objects.all(), + [ + ("George Martin"), + ("J. Tolkien"), + ], + transform=lambda x: x.name, + ordered=False, ) - self.assertQuerysetEqual(Author.objects.all(), [ - ('George Martin'), - ('J. Tolkien'), - ], transform=lambda x: x.name, ordered=False) def test_replace_with_default_arg(self): # The default replacement is an empty string. - qs = Author.objects.annotate(same_name=Replace(F('name'), Value('R. R. '))) - self.assertQuerysetEqual(qs, [ - ('George R. R. Martin', 'George Martin'), - ('J. R. R. Tolkien', 'J. Tolkien'), - ], transform=lambda x: (x.name, x.same_name), ordered=False) + qs = Author.objects.annotate(same_name=Replace(F("name"), Value("R. R. "))) + self.assertQuerysetEqual( + qs, + [ + ("George R. R. Martin", "George Martin"), + ("J. R. R. Tolkien", "J. Tolkien"), + ], + transform=lambda x: (x.name, x.same_name), + ordered=False, + ) |
