@@ -191,8 +191,7 @@ def setUp(self):
191191class TestSmallFlatDocCreation (SmallFlatDocTest , TestCase ):
192192 def do_task (self ):
193193 for doc in self .documents :
194- model = SmallFlatModel (** doc )
195- model .save ()
194+ SmallFlatModel .objects .create (** doc )
196195
197196 def after (self ):
198197 SmallFlatModel .objects .all ().delete ()
@@ -202,8 +201,7 @@ class TestSmallFlatDocUpdate(SmallFlatDocTest, TestCase):
202201 def setUp (self ):
203202 super ().setUp ()
204203 for doc in self .documents :
205- model = SmallFlatModel (** doc )
206- model .save ()
204+ SmallFlatModel .objects .create (** doc )
207205 self .models = list (SmallFlatModel .objects .all ())
208206 self .data_size = len (encode ({"field1" : "updated_value0" })) * NUM_DOCS
209207 self .iteration = 0
@@ -223,8 +221,7 @@ def setUp(self):
223221 super ().setUp ()
224222 self .ids = []
225223 for doc in self .documents :
226- model = SmallFlatModel (** doc )
227- model .save ()
224+ model = SmallFlatModel .objects .create (** doc )
228225 self .ids .append (model .id )
229226
230227 def do_task (self ):
@@ -273,8 +270,7 @@ def setUp(self):
273270class TestLargeFlatDocCreation (LargeFlatDocTest , TestCase ):
274271 def do_task (self ):
275272 for doc in self .documents :
276- model = LargeFlatModel (** doc )
277- model .save ()
273+ LargeFlatModel .objects .create (** doc )
278274
279275 def after (self ):
280276 LargeFlatModel .objects .all ().delete ()
@@ -284,8 +280,7 @@ class TestLargeFlatDocUpdate(LargeFlatDocTest, TestCase):
284280 def setUp (self ):
285281 super ().setUp ()
286282 for doc in self .documents :
287- model = LargeFlatModel (** doc )
288- model .save ()
283+ LargeFlatModel .objects .create (** doc )
289284 self .models = list (LargeFlatModel .objects .all ())
290285 self .data_size = len (encode ({"field1" : "updated_value0" })) * NUM_DOCS
291286 self .iteration = 0
0 commit comments