diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2021-06-17 11:46:37 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-06-17 11:46:37 +0200 |
| commit | 8ef3341a472909f223ec0f678f11f136f55c1406 (patch) | |
| tree | 162cbd7f7a0e3bd87c2e3b5fe61f70f25ec951c5 /test/test_optim.py | |
| parent | 2dbeeda9308029a8e8db56bed07d48f4d5718efb (diff) | |
[MRG] Speedup tests (#262)
* speedup tests
* add color to tests and timings
* add test unbalanced
* stupid missing -
Diffstat (limited to 'test/test_optim.py')
| -rw-r--r-- | test/test_optim.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/test/test_optim.py b/test/test_optim.py index 48de38a..fd194c2 100644 --- a/test/test_optim.py +++ b/test/test_optim.py @@ -37,8 +37,8 @@ def test_conditional_gradient(): np.testing.assert_allclose(b, G.sum(0)) -def test_conditional_gradient2(): - n = 1000 # nb samples +def test_conditional_gradient_itermax(): + n = 100 # nb samples mu_s = np.array([0, 0]) cov_s = np.array([[1, 0], [0, 1]]) @@ -63,7 +63,7 @@ def test_conditional_gradient2(): reg = 1e-1 - G, log = ot.optim.cg(a, b, M, reg, f, df, numItermaxEmd=200000, + G, log = ot.optim.cg(a, b, M, reg, f, df, numItermaxEmd=10000, verbose=True, log=True) np.testing.assert_allclose(a, G.sum(1)) |