summaryrefslogtreecommitdiff
path: root/test/test_ot.py
diff options
context:
space:
mode:
authorNicolas Courty <ncourty@irisa.fr>2020-04-23 13:03:28 +0200
committerGitHub <noreply@github.com>2020-04-23 13:03:28 +0200
commitef12867f1425ee86b3cfddef4287b52d46114e83 (patch)
tree38e023c5561b1669f4d8e602feb6728f51e1b359 /test/test_ot.py
parentbacb0b992aa4e1ba7e5fd0beb0bf9617c801f833 (diff)
[WIP] Issue with sparse emd and adding tests on macos (#158)
* First commit-warning removal * remove dense feature * pep8 * pep8 * EMD.h * pep8 again * tic toc tolerance Co-authored-by: RĂ©mi Flamary <remi.flamary@gmail.com>
Diffstat (limited to 'test/test_ot.py')
-rw-r--r--test/test_ot.py26
1 files changed, 0 insertions, 26 deletions
diff --git a/test/test_ot.py b/test/test_ot.py
index 0f1357f..b7306f6 100644
--- a/test/test_ot.py
+++ b/test/test_ot.py
@@ -170,27 +170,6 @@ def test_emd_empty():
np.testing.assert_allclose(w, 0)
-def test_emd_sparse():
- n = 100
- rng = np.random.RandomState(0)
-
- x = rng.randn(n, 2)
- x2 = rng.randn(n, 2)
-
- M = ot.dist(x, x2)
-
- G = ot.emd([], [], M, dense=True)
-
- Gs = ot.emd([], [], M, dense=False)
-
- ws = ot.emd2([], [], M, dense=False)
-
- # check G is the same
- np.testing.assert_allclose(G, Gs.todense())
- # check value
- np.testing.assert_allclose(Gs.multiply(M).sum(), ws, rtol=1e-6)
-
-
def test_emd2_multi():
n = 500 # nb bins
@@ -222,12 +201,7 @@ def test_emd2_multi():
emdn = ot.emd2(a, b, M)
ot.toc('multi proc : {} s')
- ot.tic()
- emdn2 = ot.emd2(a, b, M, dense=False)
- ot.toc('multi proc : {} s')
-
np.testing.assert_allclose(emd1, emdn)
- np.testing.assert_allclose(emd1, emdn2, rtol=1e-6)
# emd loss multipro proc with log
ot.tic()