diff options
author | Romain Tavenard <romain.tavenard@univ-rennes2.fr> | 2019-07-01 11:06:26 +0200 |
---|---|---|
committer | Romain Tavenard <romain.tavenard@univ-rennes2.fr> | 2019-07-01 11:06:26 +0200 |
commit | b05d315b0994d328029d4a4fc082f6994e7f06d1 (patch) | |
tree | d3b4c38aac983ad770dd53937d51cd2a3141392c /test/test_gpu.py | |
parent | 93a74fe4d477e1735e9ce21ee4113281f58b4dcf (diff) |
Moved GPU doctests to test_gpu for tests not to fail if no GPU available
Diffstat (limited to 'test/test_gpu.py')
-rw-r--r-- | test/test_gpu.py | 10 |
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) |