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/postgres_tests/test_bulk_update.py | |
| parent | e73ce08888e6f34d3f050377cfd2fbb733be94a9 (diff) | |
[4.0.x] Refs #33476 -- Reformatted code with Black.
Backport of 9c19aff7c7561e3a82978a272ecdaad40dda5c00 from main.
Diffstat (limited to 'tests/postgres_tests/test_bulk_update.py')
| -rw-r--r-- | tests/postgres_tests/test_bulk_update.py | 37 |
1 files changed, 25 insertions, 12 deletions
diff --git a/tests/postgres_tests/test_bulk_update.py b/tests/postgres_tests/test_bulk_update.py index da5aee0f70..f0b473efa7 100644 --- a/tests/postgres_tests/test_bulk_update.py +++ b/tests/postgres_tests/test_bulk_update.py @@ -2,8 +2,12 @@ from datetime import date from . import PostgreSQLTestCase from .models import ( - HStoreModel, IntegerArrayModel, NestedIntegerArrayModel, - NullableIntegerArrayModel, OtherTypesArrayModel, RangesModel, + HStoreModel, + IntegerArrayModel, + NestedIntegerArrayModel, + NullableIntegerArrayModel, + OtherTypesArrayModel, + RangesModel, ) try: @@ -15,19 +19,28 @@ except ImportError: class BulkSaveTests(PostgreSQLTestCase): def test_bulk_update(self): test_data = [ - (IntegerArrayModel, 'field', [], [1, 2, 3]), - (NullableIntegerArrayModel, 'field', [1, 2, 3], None), - (NestedIntegerArrayModel, 'field', [], [[1, 2, 3]]), - (HStoreModel, 'field', {}, {1: 2}), - (RangesModel, 'ints', None, NumericRange(lower=1, upper=10)), - (RangesModel, 'dates', None, DateRange(lower=date.today(), upper=date.today())), - (OtherTypesArrayModel, 'ips', [], ['1.2.3.4']), - (OtherTypesArrayModel, 'json', [], [{'a': 'b'}]) + (IntegerArrayModel, "field", [], [1, 2, 3]), + (NullableIntegerArrayModel, "field", [1, 2, 3], None), + (NestedIntegerArrayModel, "field", [], [[1, 2, 3]]), + (HStoreModel, "field", {}, {1: 2}), + (RangesModel, "ints", None, NumericRange(lower=1, upper=10)), + ( + RangesModel, + "dates", + None, + DateRange(lower=date.today(), upper=date.today()), + ), + (OtherTypesArrayModel, "ips", [], ["1.2.3.4"]), + (OtherTypesArrayModel, "json", [], [{"a": "b"}]), ] for Model, field, initial, new in test_data: with self.subTest(model=Model, field=field): - instances = Model.objects.bulk_create(Model(**{field: initial}) for _ in range(20)) + instances = Model.objects.bulk_create( + Model(**{field: initial}) for _ in range(20) + ) for instance in instances: setattr(instance, field, new) Model.objects.bulk_update(instances, [field]) - self.assertSequenceEqual(Model.objects.filter(**{field: new}), instances) + self.assertSequenceEqual( + Model.objects.filter(**{field: new}), instances + ) |
