Moved GPU doctests to test_gpu for tests not to fail if no GPU available

1 files changed, 10 insertions, 0 deletions

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
@@ -16,6 +16,16 @@ except ImportError:
 
 
 @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():
 
     rng = np.random.RandomState(0)