diff options
Diffstat (limited to 'test/test_plot.py')
-rw-r--r-- | test/test_plot.py | 9 |
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]) |