Function test_stimp_raw_mp

tests/test_aamp_stimp.py:201–228

Check pan.P_ attribute for raw matrix profile

(T)

@pytest.mark.parametrize("T", T)
def test_stimp_raw_mp(T):
    """
    Check pan.P_ attribute for raw matrix profile
    """
    percentage = 1.0
    min_m = 3
    n = 5

    pan = aamp_stimp(
        T,
        min_m=min_m,
        max_m=None,
        step=1,
        percentage=percentage,
        pre_scraamp=False,
    )

    for i in range(n):
        pan.update()

    for idx, m in enumerate(pan.M_[:n]):
        zone = int(np.ceil(m / 4))
        ref_P_ = naive.aamp(T, m, T_B=None, exclusion_zone=zone)[:, 0]
        cmp_P_ = pan.P_[idx]

        naive.replace_inf(ref_P_)
        naive.replace_inf(cmp_P_)
        npt.assert_almost_equal(ref_P_, cmp_P_)


@pytest.mark.filterwarnings("ignore:numpy.dtype size changed")