| 295 | |
| 296 | @pytest.mark.parametrize("T, m", test_data) |
| 297 | def test_mstump_wrapper(T, m): |
| 298 | excl_zone = int(np.ceil(m / 4)) |
| 299 | |
| 300 | ref_P, ref_I = naive.mstump(T, m, excl_zone) |
| 301 | comp_P, comp_I = mstump(T, m) |
| 302 | |
| 303 | npt.assert_almost_equal(ref_P, comp_P) |
| 304 | npt.assert_almost_equal(ref_I, comp_I) |
| 305 | |
| 306 | df = pd.DataFrame(T.T) |
| 307 | comp_P, comp_I = mstump(df, m) |
| 308 | |
| 309 | npt.assert_almost_equal(ref_P, comp_P) |
| 310 | npt.assert_almost_equal(ref_I, comp_I) |
| 311 | |
| 312 | df = pl.DataFrame(T.T) |
| 313 | comp_P, comp_I = mstump(df, m) |
| 314 | |
| 315 | npt.assert_almost_equal(ref_P, comp_P) |
| 316 | npt.assert_almost_equal(ref_I, comp_I) |
| 317 | |
| 318 | |
| 319 | @pytest.mark.parametrize("T, m", test_data) |