| 219 | |
| 220 | @pytest.mark.parametrize("T, m", test_data) |
| 221 | def test_maamp_include_discords(T, m): |
| 222 | for width in range(T.shape[0]): |
| 223 | for i in range(T.shape[0] - width): |
| 224 | include = np.asarray(range(i, i + width + 1)) |
| 225 | |
| 226 | excl_zone = int(np.ceil(m / 4)) |
| 227 | |
| 228 | ref_P, ref_I = naive.maamp(T, m, excl_zone, include, discords=True) |
| 229 | comp_P, comp_I = maamp(T, m, include, discords=True) |
| 230 | |
| 231 | npt.assert_almost_equal(ref_P, comp_P) |
| 232 | npt.assert_almost_equal(ref_I, comp_I) |
| 233 | |
| 234 | |
| 235 | @pytest.mark.parametrize("T, m", test_data) |