summaryrefslogtreecommitdiff
path: root/tests/postgres_tests/test_bulk_update.py
diff options
context:
space:
mode:
authordjango-bot <ops@djangoproject.com>2022-02-03 20:24:19 +0100
committerMariusz Felisiak <felisiak.mariusz@gmail.com>2022-02-07 20:37:05 +0100
commit9c19aff7c7561e3a82978a272ecdaad40dda5c00 (patch)
treef0506b668a013d0063e5fba3dbf4863b466713ba /tests/postgres_tests/test_bulk_update.py
parentf68fa8b45dfac545cfc4111d4e52804c86db68d3 (diff)
Refs #33476 -- Reformatted code with Black.
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
+ )