diff options
author | Rémi Flamary <remi.flamary@gmail.com> | 2019-12-02 13:34:55 +0100 |
---|---|---|
committer | Rémi Flamary <remi.flamary@gmail.com> | 2019-12-02 13:34:55 +0100 |
commit | 7371b2f4f931db8f67ec2967253be8d95ff9fe80 (patch) | |
tree | dc64502d21d2493b07b62d2ea4c29e48058eb670 | |
parent | 84384dd9e5dc78ed5cc867a53bd1de31c05d77fc (diff) |
add test emd2
-rw-r--r-- | test/test_ot.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/test/test_ot.py b/test/test_ot.py index 507d188..48ea87f 100644 --- a/test/test_ot.py +++ b/test/test_ot.py @@ -171,7 +171,12 @@ 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) # emd loss multipro proc with log ot.tic() |