| 534 | |
| 535 | |
| 536 | def test_stumpi_constant_subsequence_self_join(): |
| 537 | m = 3 |
| 538 | zone = int(np.ceil(m / 4)) |
| 539 | |
| 540 | seed = np.random.randint(100000) |
| 541 | np.random.seed(seed) |
| 542 | |
| 543 | T = np.concatenate((np.zeros(20, dtype=np.float64), np.ones(10, dtype=np.float64))) |
| 544 | stream = stumpi(T, m, egress=False) |
| 545 | for i in range(34): |
| 546 | t = np.random.rand() |
| 547 | stream.update(t) |
| 548 | |
| 549 | comp_P = stream.P_ |
| 550 | # comp_I = stream.I_ |
| 551 | |
| 552 | ref_mp = naive.stump(stream.T_, m, exclusion_zone=zone, row_wise=True) |
| 553 | ref_P = ref_mp[:, 0] |
| 554 | # ref_I = ref_mp[:, 1] |
| 555 | |
| 556 | naive.replace_inf(ref_P) |
| 557 | naive.replace_inf(comp_P) |
| 558 | |
| 559 | npt.assert_almost_equal(ref_P, comp_P) |
| 560 | # npt.assert_almost_equal(ref_I, comp_I) |
| 561 | |
| 562 | np.random.seed(seed) |
| 563 | T = np.concatenate((np.zeros(20, dtype=np.float64), np.ones(10, dtype=np.float64))) |
| 564 | T = pd.Series(T) |
| 565 | stream = stumpi(T, m, egress=False) |
| 566 | for i in range(34): |
| 567 | t = np.random.rand() |
| 568 | stream.update(t) |
| 569 | |
| 570 | comp_P = stream.P_ |
| 571 | # comp_I = stream.I_ |
| 572 | |
| 573 | naive.replace_inf(comp_P) |
| 574 | |
| 575 | npt.assert_almost_equal(ref_P, comp_P) |
| 576 | # npt.assert_almost_equal(ref_I, comp_I) |
| 577 | |
| 578 | |
| 579 | def test_stumpi_constant_subsequence_self_join_egress(): |