1 files changed, 6 insertions, 3 deletions
diff --git a/test/test_plot.py b/test/test_plot.py
index 8916a85..69789fa 100644
--- a/test/test_plot.py
+++ b/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])