Function test_fit_xrf_map2

pyxrf/core/tests/test_map_processing.py:910–943

Basic functionality of `fit_xrf_map`. Tests are run using global Dask clients to inprove testing speed.

()

def test_fit_xrf_map2():
    """
    Basic functionality of `fit_xrf_map`.
    Tests are run using global Dask clients to inprove testing speed.
    """

    dataset_params = {"n_data_dimensions": (20, 20)}
    add_pts_before, add_pts_after = 15, 10
    use_snip = False

    ft = _FitXRFMapTesting(
        dataset_params=dataset_params,
        use_snip=use_snip,
        add_pts_before=add_pts_before,
        add_pts_after=add_pts_after,
    )

    # Just run the test with input data represented as numpy array
    data = ft.data_input

    # Run fitting
    data_out = fit_xrf_map(
        data,
        data_sel_indices=ft.data_sel_indices,
        matv=ft.spectra,
        snip_param=ft.snip_param,
        use_snip=use_snip,
        chunk_pixels=10,
        n_chunks_min=4,
        progress_bar=None,
        client=None,
    )

    ft.verify_fit_output(data_out=data_out, snip_param=ft.snip_param)


# fmt: off