diff options
author | Gard Spreemann <gspr@nonempty.org> | 2023-01-20 13:51:54 +0100 |
---|---|---|
committer | Gard Spreemann <gspr@nonempty.org> | 2023-01-20 13:51:54 +0100 |
commit | 58db7d48d3315d5b4112e2ceaeee8f8c137f9de1 (patch) | |
tree | a755190f4ab6a8a59b642f81ede39e14e7ae1811 /src/python/test/test_wasserstein_distance.py | |
parent | 966b33d9da2b8044c7abc111e7b80ce7c83afeaa (diff) | |
parent | 688cbc5c29d0a6aaf233eee185a06cef5cb2e745 (diff) |
Merge branch 'dfsg/latest' into debian/sid
Diffstat (limited to 'src/python/test/test_wasserstein_distance.py')
-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): |