| 81 | @pytest.mark.filterwarnings("ignore:\\s+Port 8787 is already in use:UserWarning") |
| 82 | @pytest.mark.parametrize("T, m", test_data) |
| 83 | def test_maamped_include_discords(T, m, dask_cluster): |
| 84 | with Client(dask_cluster) as dask_client: |
| 85 | for width in range(T.shape[0]): |
| 86 | for i in range(T.shape[0] - width): |
| 87 | include = np.asarray(range(i, i + width + 1)) |
| 88 | |
| 89 | excl_zone = int(np.ceil(m / 4)) |
| 90 | |
| 91 | ref_P, ref_I = naive.maamp(T, m, excl_zone, include, discords=True) |
| 92 | comp_P, comp_I = maamped(dask_client, T, m, include, discords=True) |
| 93 | |
| 94 | npt.assert_almost_equal(ref_P, comp_P) |
| 95 | npt.assert_almost_equal(ref_I, comp_I) |
| 96 | |
| 97 | |
| 98 | @pytest.mark.filterwarnings("ignore:\\s+Port 8787 is already in use:UserWarning") |