(T_A, T_B)
| 29 | |
| 30 | @pytest.mark.parametrize("T_A, T_B", test_data) |
| 31 | def test_aamp_self_join(T_A, T_B): |
| 32 | m = 3 |
| 33 | for p in [1.0, 2.0, 3.0]: |
| 34 | ref_mp = naive.aamp(T_B, m, p=p) |
| 35 | comp_mp = aamp(T_B, m, p=p) |
| 36 | naive.replace_inf(ref_mp) |
| 37 | naive.replace_inf(comp_mp) |
| 38 | npt.assert_almost_equal(ref_mp, comp_mp) |
| 39 | |
| 40 | comp_mp = aamp(pd.Series(T_B), m, p=p) |
| 41 | naive.replace_inf(comp_mp) |
| 42 | npt.assert_almost_equal(ref_mp, comp_mp) |
| 43 | |
| 44 | |
| 45 | @pytest.mark.parametrize("T_A, T_B", test_data) |