diff options
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 + ) |
