test plot with no X

Author: rflamary

test/test_plot.py

@@ -1,13 +1,14 @@
 
 
-import ot
 import numpy as np
-
-# import pytest
+import matplotlib
+matplotlib.use('Agg')
 
 
 def test_plot1D_mat():
 
+    import ot
+
     n = 100  # nb bins
 
     # bin positions
@@ -26,6 +27,8 @@ def test_plot1D_mat():
 
 def test_plot2D_samples_mat():
 
+    import ot
+
     n = 50  # nb samples
 
     mu_s = np.array([0, 0])