def test_mpdisted(T_A, T_B, dask_cluster):
with Client(dask_cluster) as dask_client:
m = 3
ref_mpdist = naive.mpdist(T_A, T_B, m)
comp_mpdist = mpdisted(dask_client, T_A, T_B, m)
npt.assert_almost_equal(ref_mpdist, comp_mpdist)
def test_mpdist_custom_func(T_A, T_B, k):
m = 3
percentage = 0.05
n_A = T_A.shape[0]
n_B = T_B.shape[0]
partial_k_func = partial(some_func,
m=m,
percentage=percentage,
n_A=n_A,
n_B=n_B)
ref_mpdist = naive.mpdist(T_A, T_B, m)
comp_mpdist = _mpdist(T_A, T_B, m, custom_func=partial_k_func)
npt.assert_almost_equal(ref_mpdist, comp_mpdist)
def test_gpu_mpdist(T_A, T_B):
m = 3
ref_mpdist = naive.mpdist(T_A, T_B, m)
comp_mpdist = gpu_mpdist(T_A, T_B, m)
npt.assert_almost_equal(ref_mpdist, comp_mpdist)
def test_mpdist_k(T_A, T_B, k):
m = 3
ref_mpdist = naive.mpdist(T_A, T_B, m, k=k)
comp_mpdist = mpdist(T_A, T_B, m, k=k)
npt.assert_almost_equal(ref_mpdist, comp_mpdist)
def test_mpdist_percentage(T_A, T_B, percentage):
m = 3
ref_mpdist = naive.mpdist(T_A, T_B, m, percentage=percentage)
comp_mpdist = mpdist(T_A, T_B, m, percentage=percentage)
npt.assert_almost_equal(ref_mpdist, comp_mpdist)
