From b05d315b0994d328029d4a4fc082f6994e7f06d1 Mon Sep 17 00:00:00 2001
From: Romain Tavenard <romain.tavenard@univ-rennes2.fr>
Date: Mon, 1 Jul 2019 11:06:26 +0200
Subject: Moved GPU doctests to test_gpu for tests not to fail if no GPU
 available

---
 test/test_gpu.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'test/test_gpu.py')

diff --git a/test/test_gpu.py b/test/test_gpu.py
index 6b7fdd4..47b8b6d 100644
--- a/test/test_gpu.py
+++ b/test/test_gpu.py
@@ -15,6 +15,16 @@ except ImportError:
     nogpu = True
 
 
+@pytest.mark.skipif(nogpu, reason="No GPU available")
+def test_gpu_old_doctests():
+    a = [.5, .5]
+    b = [.5, .5]
+    M = [[0., 1.], [1., 0.]]
+    G = ot.sinkhorn(a, b, M, 1)
+    np.testing.assert_allclose(G, np.array([[0.36552929, 0.13447071],
+                                     [0.13447071, 0.36552929]]))
+
+
 @pytest.mark.skipif(nogpu, reason="No GPU available")
 def test_gpu_dist():
 
-- 
cgit v1.2.3