diff options
author | Vincent Rouvreau <10407034+VincentRouvreau@users.noreply.github.com> | 2022-11-21 15:02:01 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-11-21 15:02:01 +0100 |
commit | 79a713843d79bc99deecd796e9ccfd2b5a55da01 (patch) | |
tree | 1455348da54934ab340c367ae52b7a0cc7965805 | |
parent | efd1a1270f15870e015701a31e3848aa09a27c05 (diff) | |
parent | 1d90a301af29608da143be70b75d5a4fddc88e57 (diff) |
Merge pull request #746 from mglisse/warnwass
Catch warnings in wasserstein test
-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): |