diff options
Diffstat (limited to 'test/test_partial.py')
-rwxr-xr-x | test/test_partial.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/test/test_partial.py b/test/test_partial.py index 121f345..3571e2a 100755 --- a/test/test_partial.py +++ b/test/test_partial.py @@ -129,9 +129,9 @@ def test_partial_wasserstein(): # check constratints np.testing.assert_equal( - G.sum(1) <= p, [True] * len(p)) # cf convergence wasserstein + G.sum(1) - p <= 1e-5, [True] * len(p)) # cf convergence wasserstein np.testing.assert_equal( - G.sum(0) <= q, [True] * len(q)) # cf convergence wasserstein + G.sum(0) - q <= 1e-5, [True] * len(q)) # cf convergence wasserstein np.testing.assert_allclose( np.sum(G), m, atol=1e-04) |