| 524 | |
| 525 | |
| 526 | def test_aampi_constant_subsequence_self_join(): |
| 527 | m = 3 |
| 528 | |
| 529 | seed = np.random.randint(100000) |
| 530 | np.random.seed(seed) |
| 531 | |
| 532 | T = np.concatenate((np.zeros(20, dtype=np.float64), np.ones(10, dtype=np.float64))) |
| 533 | stream = aampi(T, m, egress=False) |
| 534 | for i in range(34): |
| 535 | t = np.random.rand() |
| 536 | stream.update(t) |
| 537 | |
| 538 | comp_P = stream.P_ |
| 539 | # comp_I = stream.I_ |
| 540 | |
| 541 | ref_mp = naive.aamp(stream.T_, m) |
| 542 | ref_P = ref_mp[:, 0] |
| 543 | # ref_I = ref_mp[:, 1] |
| 544 | |
| 545 | naive.replace_inf(ref_P) |
| 546 | naive.replace_inf(comp_P) |
| 547 | |
| 548 | npt.assert_almost_equal(ref_P, comp_P) |
| 549 | # npt.assert_almost_equal(ref_I, comp_I) |
| 550 | |
| 551 | np.random.seed(seed) |
| 552 | T = np.concatenate((np.zeros(20, dtype=np.float64), np.ones(10, dtype=np.float64))) |
| 553 | T = pd.Series(T) |
| 554 | stream = aampi(T, m, egress=False) |
| 555 | for i in range(34): |
| 556 | t = np.random.rand() |
| 557 | stream.update(t) |
| 558 | |
| 559 | comp_P = stream.P_ |
| 560 | # comp_I = stream.I_ |
| 561 | |
| 562 | naive.replace_inf(comp_P) |
| 563 | |
| 564 | npt.assert_almost_equal(ref_P, comp_P) |
| 565 | # npt.assert_almost_equal(ref_I, comp_I) |
| 566 | |
| 567 | |
| 568 | def test_aampi_constant_subsequence_self_join_egress(): |