| 114 | @pytest.mark.filterwarnings("ignore:\\s+Port 8787 is already in use:UserWarning") |
| 115 | @pytest.mark.parametrize("T_A, T_B", test_data) |
| 116 | def test_aamped_A_B_join(T_A, T_B, dask_cluster): |
| 117 | with Client(dask_cluster) as dask_client: |
| 118 | m = 3 |
| 119 | for p in [1.0, 2.0, 3.0]: |
| 120 | ref_mp = naive.aamp(T_A, m, T_B=T_B, p=p) |
| 121 | comp_mp = aamped(dask_client, T_A, m, T_B, ignore_trivial=False, p=p) |
| 122 | naive.replace_inf(ref_mp) |
| 123 | naive.replace_inf(comp_mp) |
| 124 | npt.assert_almost_equal(ref_mp, comp_mp) |
| 125 | |
| 126 | |
| 127 | @pytest.mark.filterwarnings("ignore:numpy.dtype size changed") |