| 130 | @pytest.mark.filterwarnings("ignore:\\s+Port 8787 is already in use:UserWarning") |
| 131 | @pytest.mark.parametrize("T_A, T_B", test_data) |
| 132 | def test_aamped_A_B_join_df(T_A, T_B, dask_cluster): |
| 133 | with Client(dask_cluster) as dask_client: |
| 134 | m = 3 |
| 135 | ref_mp = naive.aamp(T_A, m, T_B=T_B) |
| 136 | comp_mp = aamped( |
| 137 | dask_client, pd.Series(T_A), m, pd.Series(T_B), ignore_trivial=False |
| 138 | ) |
| 139 | naive.replace_inf(ref_mp) |
| 140 | naive.replace_inf(comp_mp) |
| 141 | npt.assert_almost_equal(ref_mp, comp_mp) |
| 142 | |
| 143 | |
| 144 | @pytest.mark.filterwarnings("ignore:A large number of values are smaller") |