def test_emd(): xx = np.array([[1, 0, 0], [0, 1, 0], [0, 0, 1]]) cost = [] for this_x in xx: for this_y in xx: cost.append(np.sqrt(sum((this_x - this_y) ** 2))) ee = emd.emd([0, 1, 0], [1, 0, 0], list(cost)) npt.assert_almost_equal(ee, np.sqrt(2), decimal=5)
def test_emd(): xx = np.array([[1, 0, 0], [0, 1, 0], [0, 0, 1]]) cost = [] for this_x in xx: for this_y in xx: cost.append(np.sqrt(sum((this_x - this_y)**2))) ee = emd.emd([0, 1, 0], [1, 0, 0], list(cost)) npt.assert_almost_equal(ee, np.sqrt(2), decimal=5)