summaryrefslogtreecommitdiff
path: root/tests/postgres_tests/test_bulk_update.py
diff options
context:
space:
mode:
authordjango-bot <ops@djangoproject.com>2022-02-08 12:09:55 +0100
committerMariusz Felisiak <felisiak.mariusz@gmail.com>2022-02-08 12:15:38 +0100
commit6a682b38e75d4c975b4c4493565a59f1bc14397c (patch)
tree0bd9cda550bea26238656d9f120d769e8b41bb9e /tests/postgres_tests/test_bulk_update.py
parente73ce08888e6f34d3f050377cfd2fbb733be94a9 (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.py37
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
+ )