def test_consolidate_chunks_limit_error(shape, chunks, itemsize, max_mem,
chunk_limits):
with pytest.raises(ValueError, match=r"Invalid chunk_limits .*"):
consolidate_chunks(shape,
chunks,
itemsize,
max_mem,
chunk_limits=chunk_limits)
def test_consolidate_chunks_4D(shape, chunks, itemsize, max_mem, expected):
"""A realistic example."""
new_chunks = consolidate_chunks(shape, chunks, itemsize, max_mem)
assert new_chunks == expected
chunk_mem = itemsize * new_chunks[0] * new_chunks[1] * new_chunks[
2] * new_chunks[3]
assert itemsize <= max_mem
def test_consolidate_chunks_w_limits(shape, chunks, itemsize, max_mem,
chunk_limits, expected):
new_chunks = consolidate_chunks(shape,
chunks,
itemsize,
max_mem,
chunk_limits=chunk_limits)
assert new_chunks == expected
def test_consolidate_chunks_mem_error():
shape, chunks, itemsize = (8, 8), (1, 2), 4
max_mem = 7
with pytest.raises(ValueError, match=r"chunk_mem 8 > max_mem 7"):
consolidate_chunks(shape, chunks, itemsize, max_mem)
def test_consolidate_chunks(shape, chunks, itemsize, max_mem, expected):
new_chunks = consolidate_chunks(shape, chunks, itemsize, max_mem)
assert new_chunks == expected