Update tests for calc batch size to accomodate new sqlite logic

This commit is contained in:
Ethan Paul 2023-04-13 14:47:31 -04:00
parent 0d1a142e43
commit 5df974faa5
1 changed files with 5 additions and 1 deletions

View File

@ -17,7 +17,11 @@ def test_sqlite(fakedb):
data = peewee.IntegerField()
models = [TestModel(item) for item in range(500)]
# Three is just chosen as an arbitrary multiplier to ensure the value is larger than the
# sqlite variable limit
models = [
TestModel(item) for item in range(peewee_plus.SQLITE_DEFAULT_VARIABLE_LIMIT * 3)
]
assert (
peewee_plus.calc_batch_size(models) <= peewee_plus.SQLITE_DEFAULT_VARIABLE_LIMIT
)