Function test_maamp_include

tests/test_maamp.py:196–206

(T, m)

@pytest.mark.parametrize("T, m", test_data)
def test_maamp_include(T, m):
    for width in range(T.shape[0]):
        for i in range(T.shape[0] - width):
            include = np.asarray(range(i, i + width + 1))
            excl_zone = int(np.ceil(m / 4))

            ref_P, ref_I = naive.maamp(T, m, excl_zone, include)
            comp_P, comp_I = maamp(T, m, include)

            npt.assert_almost_equal(ref_P, comp_P)
            npt.assert_almost_equal(ref_I, comp_I)


@pytest.mark.parametrize("T, m", test_data)