Function test_stimped

tests/test_non_normalized_decorator.py:446–473

(T, m, dask_cluster)

@pytest.mark.filterwarnings("ignore:\\s+Port 8787 is already in use:UserWarning")
@pytest.mark.parametrize("T, m", test_data)
def test_stimped(T, m, dask_cluster):
    if T.ndim > 1:
        T = T.copy()
        T = T[0]
    n = 3
    seed = np.random.randint(100000)
    with Client(dask_cluster) as dask_client:
        np.random.seed(seed)
        ref = aamp_stimped(dask_client, T, m)
        for i in range(n):
            ref.update()

        np.random.seed(seed)
        cmp = stimped(dask_client, T, m, normalize=False)
        for i in range(n):
            cmp.update()

        # Compare raw pan
        ref_PAN = ref._PAN
        cmp_PAN = cmp._PAN

        naive.replace_inf(ref_PAN)
        naive.replace_inf(cmp_PAN)

        npt.assert_almost_equal(ref_PAN, cmp_PAN)

        # Compare transformed pan
        npt.assert_almost_equal(ref.PAN_, cmp.PAN_)


@pytest.mark.filterwarnings("ignore", category=NumbaPerformanceWarning)