test raise with pytets in test_emd_1d_emd2_1d

Author: rflamary

Makefile

@@ -42,10 +42,10 @@ pep8 :
 	flake8 examples/ ot/ test/
 
 test : FORCE pep8
-	$(PYTHON) -m pytest -v test/ --cov=ot --cov-report html:cov_html
+	$(PYTHON) -m pytest -v test/ --doctest-modules --ignore ot/gpu/  --cov=ot --cov-report html:cov_html
 
 pytest : FORCE 
-	$(PYTHON) -m pytest -v test/ --cov=ot
+	$(PYTHON) -m pytest -v test/ --doctest-modules --ignore ot/gpu/  --cov=ot
 
 uploadpypi :
 	#python setup.py register

ot/__init__.py

@@ -10,7 +10,10 @@
     :py:mod:`ot.gromov`, :py:mod:`ot.smooth`
     :py:mod:`ot.stochastic`    
 
-    The other sub-modules are not imported due to additional dependencies.
+    The following sub-modules are not imported due to additional dependencies:
+
+    - :any:`ot.dr` : depends on :code:`pymanopt` and :code:`autograd`.
+    - :any:`ot.gpu` : depends on :code:`cupy` and a CUDA GPU.
 
 """
 

test/test_ot.py

@@ -72,7 +72,8 @@ def test_emd_1d_emd2_1d():
     # check AssertionError is raised if called on non 1d arrays
     u = np.random.randn(n, 2)
     v = np.random.randn(m, 2)
-    np.testing.assert_raises(AssertionError, ot.emd_1d, u, v, [], [])
+    with pytest.raises(AssertionError):
+        ot.emd_1d(u, v, [], [])
 
 
 def test_wass_1d():