diff options
author | Rémi Flamary <remi.flamary@gmail.com> | 2020-04-24 13:59:42 +0200 |
---|---|---|
committer | Rémi Flamary <remi.flamary@gmail.com> | 2020-04-24 13:59:42 +0200 |
commit | 90bd408e86eccb03b02d57a0cd7963e0c848a1fc (patch) | |
tree | 6a2fabd4dc411953148f13cbe145c5dfbbdc96b8 /test | |
parent | c3115bce070cbf567c4dda3dfe87114166595423 (diff) |
pep8
Diffstat (limited to 'test')
-rwxr-xr-x | test/test_partial.py | 5 | ||||
-rw-r--r-- | test/test_stochastic.py | 4 | ||||
-rw-r--r-- | test/test_unbalanced.py | 2 |
3 files changed, 5 insertions, 6 deletions
diff --git a/test/test_partial.py b/test/test_partial.py index 5960e4e..b533a9c 100755 --- a/test/test_partial.py +++ b/test/test_partial.py @@ -9,6 +9,7 @@ import numpy as np import scipy as sp import ot + def test_partial_wasserstein_lagrange(): n_samples = 20 # nb samples (gaussian) @@ -29,9 +30,7 @@ def test_partial_wasserstein_lagrange(): m = 0.5 - w0, log0 = ot.partial.partial_wasserstein_lagrange(p, q, M, 1, log=True) - - + w0, log0 = ot.partial.partial_wasserstein_lagrange(p, q, M, 1, log=True) def test_partial_wasserstein(): diff --git a/test/test_stochastic.py b/test/test_stochastic.py index 8ddf485..155622c 100644 --- a/test/test_stochastic.py +++ b/test/test_stochastic.py @@ -71,7 +71,7 @@ def test_stochastic_asgd(): M = ot.dist(x, x) G, log = ot.stochastic.solve_semi_dual_entropic(u, u, M, reg, "asgd", - numItermax=numItermax, log=True) + numItermax=numItermax, log=True) # check constratints np.testing.assert_allclose( @@ -146,7 +146,7 @@ def test_stochastic_dual_sgd(): M = ot.dist(x, x) G, log = ot.stochastic.solve_dual_entropic(u, u, M, reg, batch_size, - numItermax=numItermax, log=True) + numItermax=numItermax, log=True) # check constratints np.testing.assert_allclose( diff --git a/test/test_unbalanced.py b/test/test_unbalanced.py index d5bae42..dfeaad9 100644 --- a/test/test_unbalanced.py +++ b/test/test_unbalanced.py @@ -35,7 +35,7 @@ def test_unbalanced_convergence(method): verbose=True) loss = ot.unbalanced.sinkhorn_unbalanced2(a, b, M, epsilon, reg_m, method=method, - verbose=True) + verbose=True) # check fixed point equations # in log-domain fi = reg_m / (reg_m + epsilon) |