diff options
Diffstat (limited to 'tests/bulk_create/tests.py')
| -rw-r--r-- | tests/bulk_create/tests.py | 15 |
1 files changed, 11 insertions, 4 deletions
diff --git a/tests/bulk_create/tests.py b/tests/bulk_create/tests.py index e6d85cb815..3378f0010a 100644 --- a/tests/bulk_create/tests.py +++ b/tests/bulk_create/tests.py @@ -171,11 +171,18 @@ class BulkCreateTests(TestCase): def test_explicit_batch_size(self): objs = [TwoFields(f1=i, f2=i) for i in range(0, 4)] - TwoFields.objects.bulk_create(objs, 2) - self.assertEqual(TwoFields.objects.count(), len(objs)) + num_objs = len(objs) + TwoFields.objects.bulk_create(objs, batch_size=1) + self.assertEqual(TwoFields.objects.count(), num_objs) TwoFields.objects.all().delete() - TwoFields.objects.bulk_create(objs, len(objs)) - self.assertEqual(TwoFields.objects.count(), len(objs)) + TwoFields.objects.bulk_create(objs, batch_size=2) + self.assertEqual(TwoFields.objects.count(), num_objs) + TwoFields.objects.all().delete() + TwoFields.objects.bulk_create(objs, batch_size=3) + self.assertEqual(TwoFields.objects.count(), num_objs) + TwoFields.objects.all().delete() + TwoFields.objects.bulk_create(objs, batch_size=num_objs) + self.assertEqual(TwoFields.objects.count(), num_objs) def test_empty_model(self): NoFields.objects.bulk_create([NoFields() for i in range(2)]) |