| 68 | @pytest.mark.filterwarnings("ignore:\\s+Port 8787 is already in use:UserWarning") |
| 69 | @pytest.mark.parametrize("T_A, T_B", test_data) |
| 70 | def test_stumped_self_join_df(T_A, T_B, dask_cluster): |
| 71 | with Client(dask_cluster) as dask_client: |
| 72 | m = 3 |
| 73 | zone = int(np.ceil(m / 4)) |
| 74 | ref_mp = naive.stump(T_B, m, exclusion_zone=zone) |
| 75 | comp_mp = stumped(dask_client, pd.Series(T_B), m, ignore_trivial=True) |
| 76 | naive.replace_inf(ref_mp) |
| 77 | naive.replace_inf(comp_mp) |
| 78 | npt.assert_almost_equal(ref_mp, comp_mp) |
| 79 | |
| 80 | comp_mp = stumped(dask_client, pl.Series(T_B), m, ignore_trivial=True) |
| 81 | naive.replace_inf(comp_mp) |
| 82 | npt.assert_almost_equal(ref_mp, comp_mp) |
| 83 | |
| 84 | |
| 85 | @pytest.mark.filterwarnings("ignore:numpy.dtype size changed") |