diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2020-04-24 12:39:38 +0200 |
|---|---|---|
| committer | Rémi Flamary <remi.flamary@gmail.com> | 2020-04-24 12:39:38 +0200 |
| commit | 53b063ed6b6aa15d6cb103a9304bbd169678b2e9 (patch) | |
| tree | 33f75e733e1f93c07a5d37f72f085c9722bf19c7 /test/test_unbalanced.py | |
| parent | 46523dc0956fd17e709f958ebd351e748fca0a23 (diff) | |
better coverage options verbose and log
Diffstat (limited to 'test/test_unbalanced.py')
| -rw-r--r-- | test/test_unbalanced.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/test/test_unbalanced.py b/test/test_unbalanced.py index ca1efba..d5bae42 100644 --- a/test/test_unbalanced.py +++ b/test/test_unbalanced.py @@ -31,9 +31,11 @@ def test_unbalanced_convergence(method): G, log = ot.unbalanced.sinkhorn_unbalanced(a, b, M, reg=epsilon, reg_m=reg_m, method=method, - log=True) + log=True, + verbose=True) loss = ot.unbalanced.sinkhorn_unbalanced2(a, b, M, epsilon, reg_m, - method=method) + method=method, + verbose=True) # check fixed point equations # in log-domain fi = reg_m / (reg_m + epsilon) @@ -73,7 +75,8 @@ def test_unbalanced_multiple_inputs(method): loss, log = ot.unbalanced.sinkhorn_unbalanced(a, b, M, reg=epsilon, reg_m=reg_m, method=method, - log=True) + log=True, + verbose=True) # check fixed point equations # in log-domain fi = reg_m / (reg_m + epsilon) |