(T, m)
| 123 | |
| 124 | @pytest.mark.parametrize("T, m", test_data) |
| 125 | def test_scrump(T, m): |
| 126 | if T.ndim > 1: |
| 127 | T = T.copy() |
| 128 | T = T[0] |
| 129 | |
| 130 | seed = np.random.randint(100000) |
| 131 | |
| 132 | np.random.seed(seed) |
| 133 | ref = scraamp(T, m) |
| 134 | np.random.seed(seed) |
| 135 | comp = scrump(T, m, normalize=False) |
| 136 | npt.assert_almost_equal(ref.P_, comp.P_) |
| 137 | |
| 138 | for i in range(10): |
| 139 | ref.update() |
| 140 | comp.update() |
| 141 | npt.assert_almost_equal(ref.P_, comp.P_) |
| 142 | |
| 143 | |
| 144 | @pytest.mark.parametrize("T, m", test_data) |