(T_A, T_B)
| 53 | @pytest.mark.parametrize("T_A, T_B", test_data) |
| 54 | @patch("stumpy.config.STUMPY_THREADS_PER_BLOCK", TEST_THREADS_PER_BLOCK) |
| 55 | def test_gpu_aamp_self_join(T_A, T_B): |
| 56 | m = 3 |
| 57 | zone = int(np.ceil(m / 4)) |
| 58 | for p in [1.0, 2.0, 3.0]: |
| 59 | ref_mp = naive.aamp(T_B, m, exclusion_zone=zone, p=p, row_wise=True) |
| 60 | comp_mp = gpu_aamp(T_B, m, ignore_trivial=True, p=p) |
| 61 | naive.replace_inf(ref_mp) |
| 62 | naive.replace_inf(comp_mp) |
| 63 | npt.assert_almost_equal(ref_mp, comp_mp) |
| 64 | |
| 65 | comp_mp = gpu_aamp(pd.Series(T_B), m, ignore_trivial=True, p=p) |
| 66 | naive.replace_inf(comp_mp) |
| 67 | npt.assert_almost_equal(ref_mp, comp_mp) |
| 68 | |
| 69 | |
| 70 | @pytest.mark.filterwarnings("ignore", category=NumbaPerformanceWarning) |
nothing calls this directly
no test coverage detected