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