| 216 | @pytest.mark.parametrize("m", window_size) |
| 217 | @pytest.mark.parametrize("percentages", percentages) |
| 218 | def test_scraamp_self_join_larger_window(T_A, T_B, m, percentages): |
| 219 | if len(T_B) > m: |
| 220 | zone = int(np.ceil(m / 4)) |
| 221 | |
| 222 | for percentage in percentages: |
| 223 | seed = np.random.randint(100000) |
| 224 | |
| 225 | np.random.seed(seed) |
| 226 | ref_P, ref_I, ref_left_I, ref_right_I = naive.scraamp( |
| 227 | T_B, m, T_B, percentage, zone, False, None |
| 228 | ) |
| 229 | |
| 230 | np.random.seed(seed) |
| 231 | approx = scraamp( |
| 232 | T_B, m, ignore_trivial=True, percentage=percentage, pre_scraamp=False |
| 233 | ) |
| 234 | approx.update() |
| 235 | comp_P = approx.P_ |
| 236 | comp_I = approx.I_ |
| 237 | comp_left_I = approx.left_I_ |
| 238 | comp_right_I = approx.right_I_ |
| 239 | |
| 240 | naive.replace_inf(ref_P) |
| 241 | naive.replace_inf(comp_P) |
| 242 | |
| 243 | npt.assert_almost_equal(ref_P, comp_P) |
| 244 | npt.assert_almost_equal(ref_I, comp_I) |
| 245 | npt.assert_almost_equal(ref_left_I, comp_left_I) |
| 246 | npt.assert_almost_equal(ref_right_I, comp_right_I) |
| 247 | |
| 248 | |
| 249 | @pytest.mark.parametrize("T_A, T_B", test_data) |