(T, m, dask_cluster)
| 183 | @pytest.mark.filterwarnings("ignore:\\s+Port 8787 is already in use:UserWarning") |
| 184 | @pytest.mark.parametrize("T, m", test_data) |
| 185 | def test_stumped(T, m, dask_cluster): |
| 186 | if T.ndim > 1: |
| 187 | T = T.copy() |
| 188 | T = T[0] |
| 189 | |
| 190 | with Client(dask_cluster) as dask_client: |
| 191 | ref = aamped(dask_client, T, m) |
| 192 | comp = stumped(dask_client, T, m, normalize=False) |
| 193 | npt.assert_almost_equal(ref, comp) |
| 194 | |
| 195 | |
| 196 | @pytest.mark.filterwarnings("ignore", category=NumbaPerformanceWarning) |