def test_dml_none_success_case(self): """ Tests that passing None into the batch call clears any batch object """ b = BatchQuery() q = DMLQuery(TestMultiKeyModel, batch=b) assert q._batch == b q.batch(None) assert q._batch is None
def test_dml_none_success_case(self): """ Tests that passing None into the batch call clears any batch object """ b = BatchQuery() q = DMLQuery(TestMultiKeyModel, batch=b) self.assertEqual(q._batch, b) q.batch(None) self.assertIsNone(q._batch)