| 89 | @pytest.mark.parametrize("T_A, T_B", test_data) |
| 90 | @pytest.mark.parametrize("m", window_size) |
| 91 | def test_stump_self_join_larger_window(T_A, T_B, m, dask_cluster): |
| 92 | with Client(dask_cluster) as dask_client: |
| 93 | if len(T_B) > m: |
| 94 | zone = int(np.ceil(m / 4)) |
| 95 | ref_mp = naive.stump(T_B, m, exclusion_zone=zone) |
| 96 | comp_mp = stumped(dask_client, T_B, m, ignore_trivial=True) |
| 97 | naive.replace_inf(ref_mp) |
| 98 | naive.replace_inf(comp_mp) |
| 99 | |
| 100 | npt.assert_almost_equal(ref_mp, comp_mp) |
| 101 | |
| 102 | |
| 103 | @pytest.mark.filterwarnings("ignore:numpy.dtype size changed") |