summaryrefslogtreecommitdiff
path: root/test/test_ot.py
diff options
context:
space:
mode:
authorRémi Flamary <remi.flamary@gmail.com>2018-09-28 16:04:19 +0200
committerRémi Flamary <remi.flamary@gmail.com>2018-09-28 16:04:19 +0200
commit642fd5b7707c4bdce72b883c666ffdae89fde76a (patch)
treef41f9896df48802a1b0457904fadec53c158c9c6 /test/test_ot.py
parenteaa05d1411cc8b365479fb239acf4e774c68037c (diff)
speedup tests
Diffstat (limited to 'test/test_ot.py')
-rw-r--r--test/test_ot.py12
1 files changed, 6 insertions, 6 deletions
diff --git a/test/test_ot.py b/test/test_ot.py
index 45e777a..7652394 100644
--- a/test/test_ot.py
+++ b/test/test_ot.py
@@ -70,7 +70,7 @@ def test_emd_empty():
def test_emd2_multi():
- n = 1000 # nb bins
+ n = 500 # nb bins
# bin positions
x = np.arange(n, dtype=np.float64)
@@ -78,7 +78,7 @@ def test_emd2_multi():
# Gaussian distributions
a = gauss(n, m=20, s=5) # m= mean, s= std
- ls = np.arange(20, 1000, 20)
+ ls = np.arange(20, 500, 20)
nb = len(ls)
b = np.zeros((n, nb))
for i in range(nb):
@@ -207,11 +207,11 @@ def test_warnings():
def test_dual_variables():
- n = 5000 # nb bins
- m = 6000 # nb bins
+ n = 500 # nb bins
+ m = 600 # nb bins
- mean1 = 1000
- mean2 = 1100
+ mean1 = 300
+ mean2 = 400
# bin positions
x = np.arange(n, dtype=np.float64)