(T, m)
| 234 | |
| 235 | @pytest.mark.parametrize("T, m", test_data) |
| 236 | def test_maamp_wrapper(T, m): |
| 237 | excl_zone = int(np.ceil(m / 4)) |
| 238 | |
| 239 | ref_P, ref_I = naive.maamp(T, m, excl_zone) |
| 240 | comp_P, comp_I = maamp(T, m) |
| 241 | |
| 242 | npt.assert_almost_equal(ref_P, comp_P) |
| 243 | npt.assert_almost_equal(ref_I, comp_I) |
| 244 | |
| 245 | df = pd.DataFrame(T.T) |
| 246 | comp_P, comp_I = maamp(df, m) |
| 247 | |
| 248 | npt.assert_almost_equal(ref_P, comp_P) |
| 249 | npt.assert_almost_equal(ref_I, comp_I) |
| 250 | |
| 251 | |
| 252 | @pytest.mark.parametrize("T, m", test_data) |