diff options
| author | Kilian <kilian.fatras@gmail.com> | 2019-12-09 16:35:49 +0100 |
|---|---|---|
| committer | Kilian <kilian.fatras@gmail.com> | 2019-12-09 16:35:49 +0100 |
| commit | 428b44e15591071cfcd69af365d878cfd876f9d3 (patch) | |
| tree | b5b26f14b67d57d0ac41e5a6b028e05b1f24010c /test/test_ot.py | |
| parent | 92233f79e098f1930248d815e66c0a929508af59 (diff) | |
calling ot.emd test
Diffstat (limited to 'test/test_ot.py')
| -rw-r--r-- | test/test_ot.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/test/test_ot.py b/test/test_ot.py index 1343604..25cdfd4 100644 --- a/test/test_ot.py +++ b/test/test_ot.py @@ -14,8 +14,9 @@ from ot.datasets import make_1D_gauss as gauss import pytest + def test_emd_dimension_mismatch(): - # test emd and emd2 for simple identity + # test emd and emd2 for dimension mismatch n_samples = 100 n_features = 2 rng = np.random.RandomState(0) @@ -25,9 +26,9 @@ def test_emd_dimension_mismatch(): M = ot.dist(x, x) - np.testing.assert_raises(AssertionError, emd, a, a, M) + np.testing.assert_raises(AssertionError, ot.emd, a, a, M) - np.testing.assert_raises(AssertionError, emd2, a, a, M) + np.testing.assert_raises(AssertionError, ot.emd2, a, a, M) def test_emd_emd2(): |