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_da.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_da.py')
-rw-r--r-- | test/test_da.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/test/test_da.py b/test/test_da.py index 52c6a48..44bb2e9 100644 --- a/test/test_da.py +++ b/test/test_da.py @@ -106,8 +106,8 @@ def test_sinkhorn_l1l2_transport_class(): """test_sinkhorn_transport """ - ns = 150 - nt = 200 + ns = 50 + nt = 100 Xs, ys = make_data_classif('3gauss', ns) Xt, yt = make_data_classif('3gauss2', nt) @@ -448,8 +448,8 @@ def test_mapping_transport_class(): """test_mapping_transport """ - ns = 60 - nt = 120 + ns = 20 + nt = 30 Xs, ys = make_data_classif('3gauss', ns) Xt, yt = make_data_classif('3gauss2', nt) |