def test_deterministic_gpu_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.ostinato(Ts, m)
comp_radius, comp_Ts_idx, comp_subseq_idx = gpu_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_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.ostinato(Ts, m)
comp_radius, comp_Ts_idx, comp_subseq_idx = stumpy.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)