Function test_chunk_numpy_array

pyxrf/core/tests/test_map_processing.py:234–250

Basic functionailty tests for '_chunk_xrf_map_numpy

(chunk_target, data_shape)

])
# fmt: on
def test_chunk_numpy_array(chunk_target, data_shape):
    """Basic functionailty tests for '_chunk_xrf_map_numpy"""

    data = np.random.random(data_shape)
    data_dask = _chunk_numpy_array(data, chunk_target)

    chunksize_expected = tuple(
        [min(chunk_target[0], data_shape[0]), min(chunk_target[1], data_shape[1]), *data_shape[2:]]
    )

    assert data_dask.shape == data.shape, "The shape of the original and chunked array don't match"
    assert (
        data_dask.chunksize == chunksize_expected
    ), "The chunk size of the Dask array doesn't match the desired chunk size"
    npt.assert_array_equal(
        data_dask.compute(), data, err_msg="The chunked array is different from the original array"
    )


# fmt: off