Function test_compute_total_spectrum2

pyxrf/core/tests/test_map_processing.py:526–543

Create an instance of Dask client in the 'compute_total_spectrum' function to test if it works

(tmpdir)

def test_compute_total_spectrum2(tmpdir):
    """Create an instance of Dask client in the 'compute_total_spectrum' function to test if it works"""

    # Start with dask array
    data_shape = (7, 12, 20)
    data_dask = da.random.random(data_shape, chunks=(2, 3, 4))

    data_numpy = data_dask.compute()
    total_spectrum_expected = np.sum(np.sum(data_numpy, axis=0), axis=0)

    data = _create_xrf_data(data_dask, "dask_array", tmpdir=tmpdir)

    # Run computations without the progress bar
    total_spectrum = compute_total_spectrum(data, chunk_pixels=12)

    npt.assert_array_almost_equal(
        total_spectrum, total_spectrum_expected, err_msg="Total spectrum was computed incorrectly"
    )


# fmt: off