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

Author: rtavenar

ot/gpu/bregman.py

@@ -70,17 +70,6 @@ def sinkhorn_knopp(a, b, M, reg, numItermax=1000, stopThr=1e-9,
     log : dict
         log dictionary return only if log==True in parameters
 
-    Examples
-    --------
-
-    >>> import ot
-    >>> a=[.5,.5]
-    >>> b=[.5,.5]
-    >>> M=[[0.,1.],[1.,0.]]
-    >>> ot.sinkhorn(a,b,M,1)
-    array([[ 0.36552929,  0.13447071],
-           [ 0.13447071,  0.36552929]])
-
 
     References
     ----------

test/test_gpu.py

@@ -15,6 +15,16 @@
     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():