(T, m)
| 210 | |
| 211 | @pytest.mark.parametrize("T, m", test_data) |
| 212 | def test_mdl(T, m): |
| 213 | subseq_idx = np.full(T.shape[0], 1) |
| 214 | nn_idx = np.full(T.shape[0], 4) |
| 215 | |
| 216 | ref_MDL, ref_S = naive.mdl(T, m, subseq_idx, nn_idx) |
| 217 | comp_MDL, comp_S = mdl(T, m, subseq_idx, nn_idx) |
| 218 | npt.assert_almost_equal(ref_MDL, comp_MDL) |
| 219 | |
| 220 | for ref, cmp in zip(ref_S, comp_S): |
| 221 | npt.assert_almost_equal(ref, cmp) |
| 222 | |
| 223 | |
| 224 | def test_naive_mstump(): |