def test_multi_mass(T, m): trivial_idx = 2 Q = T[:, trivial_idx:trivial_idx + m] ref = naive.multi_mass(Q, T, m) M_T, Σ_T = core.compute_mean_std(T, m) comp = _multi_mass(Q, T, m, M_T, Σ_T, M_T[:, trivial_idx], Σ_T[:, trivial_idx]) npt.assert_almost_equal(ref, comp, decimal=config.STUMPY_TEST_PRECISION)
def test_multi_mass_seeded(): np.random.seed(5) T = np.random.uniform(-1000, 1000, [3, 10]).astype(np.float64) m = 5 trivial_idx = 2 Q = T[:, trivial_idx:trivial_idx + m] ref = naive.multi_mass(Q, T, m) M_T, Σ_T = core.compute_mean_std(T, m) comp = _multi_mass(Q, T, m, M_T, Σ_T, M_T[:, trivial_idx], Σ_T[:, trivial_idx]) npt.assert_almost_equal(ref, comp, decimal=config.STUMPY_TEST_PRECISION)