def test_deterministic_gpu_aamp_ostinato(seed): m = 50 np.random.seed(seed) Ts = [np.random.rand(n) for n in [64, 128, 256]] ref_radius, ref_Ts_idx, ref_subseq_idx = naive.aamp_ostinato(Ts, m) comp_radius, comp_Ts_idx, comp_subseq_idx = gpu_aamp_ostinato(Ts, m) npt.assert_almost_equal(ref_radius, comp_radius) npt.assert_almost_equal(ref_Ts_idx, comp_Ts_idx) npt.assert_almost_equal(ref_subseq_idx, comp_subseq_idx)
def test_deterministic_ostinato(seed): m = 50 np.random.seed(seed) Ts = [np.random.rand(n) for n in [64, 128, 256]] for p in [1.0, 2.0, 3.0]: ref_radius, ref_Ts_idx, ref_subseq_idx = naive.aamp_ostinato(Ts, m, p=p) comp_radius, comp_Ts_idx, comp_subseq_idx = stumpy.aamp_ostinato(Ts, m, p=p) npt.assert_almost_equal(ref_radius, comp_radius) npt.assert_almost_equal(ref_Ts_idx, comp_Ts_idx) npt.assert_almost_equal(ref_subseq_idx, comp_subseq_idx)
def test_deterministic_ostinatoed(seed, dask_cluster): with Client(dask_cluster) as dask_client: m = 50 np.random.seed(seed) Ts = [np.random.rand(n) for n in [64, 128, 256]] ref_radius, ref_Ts_idx, ref_subseq_idx = naive.aamp_ostinato(Ts, m) comp_radius, comp_Ts_idx, comp_subseq_idx = stumpy.aamp_ostinatoed( dask_client, Ts, m) npt.assert_almost_equal(ref_radius, comp_radius) npt.assert_almost_equal(ref_Ts_idx, comp_Ts_idx) npt.assert_almost_equal(ref_subseq_idx, comp_subseq_idx)