diff options
author | Marc Glisse <marc.glisse@inria.fr> | 2022-11-18 16:19:29 +0100 |
---|---|---|
committer | Marc Glisse <marc.glisse@inria.fr> | 2022-11-18 16:19:29 +0100 |
commit | 1d90a301af29608da143be70b75d5a4fddc88e57 (patch) | |
tree | f86fc99587d4dc644096683bf87eaf0fcf5f7deb /src | |
parent | 94e2b090219233be602ed9a59994c2aced288b3a (diff) |
Catch warnings in wasserstein test
Diffstat (limited to 'src')
-rwxr-xr-x | src/python/test/test_wasserstein_distance.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/src/python/test/test_wasserstein_distance.py b/src/python/test/test_wasserstein_distance.py index 3a004d77..a76b6ce7 100755 --- a/src/python/test/test_wasserstein_distance.py +++ b/src/python/test/test_wasserstein_distance.py @@ -90,10 +90,11 @@ def test_get_essential_parts(): def test_warn_infty(): - assert _warn_infty(matching=False)==np.inf - c, m = _warn_infty(matching=True) - assert (c == np.inf) - assert (m is None) + with pytest.warns(UserWarning): + assert _warn_infty(matching=False)==np.inf + c, m = _warn_infty(matching=True) + assert (c == np.inf) + assert (m is None) def _basic_wasserstein(wasserstein_distance, delta, test_infinity=True, test_matching=True): |