diff options
| author | tlacombe <lacombe1993@gmail.com> | 2019-09-23 11:14:24 +0200 |
|---|---|---|
| committer | tlacombe <lacombe1993@gmail.com> | 2019-09-23 11:14:24 +0200 |
| commit | 1b007fc59f08bd01e1521eb1c0773b598bdf158b (patch) | |
| tree | 6a5169be2dd87c28a27cd0054a8d0cd3b251c178 /src/python/test/test_wasserstein_distance.py | |
| parent | 36d82a6ffe7c099da9241f7268637feaeef6bf55 (diff) | |
wasserstein distance added on fork
Diffstat (limited to 'src/python/test/test_wasserstein_distance.py')
| -rwxr-xr-x | src/python/test/test_wasserstein_distance.py | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/src/python/test/test_wasserstein_distance.py b/src/python/test/test_wasserstein_distance.py new file mode 100755 index 00000000..a5f7cf77 --- /dev/null +++ b/src/python/test/test_wasserstein_distance.py @@ -0,0 +1,22 @@ +import gudhi + +""" This file is part of the Gudhi Library - https://gudhi.inria.fr/ - which is released under MIT. + See file LICENSE or go to https://gudhi.inria.fr/licensing/ for full license details. + Author(s): Theo Lacombe + + Copyright (C) 2016 Inria + + Modification(s): + - YYYY/MM Author: Description of the modification +""" + +__author__ = "Theo Lacombe" +__copyright__ = "Copyright (C) 2016 Inria" +__license__ = "MIT" + + +def test_basic_bottleneck(): + diag1 = np.array([[2.7, 3.7], [9.6, 14.0], [34.2, 34.974]]) + diag2 = np.array([[2.8, 4.45], [9.5, 14.1]]) + + assert gudhi.wasserstein_distance(diag1, diag2) == 1.4453593023967701 |