Protect matplotlib import in tests and tag.

We tag matplotlib tests with `@attr(‘matplotlib’)` so we can skip them.
However, we still need to test the imports.

Author: theengineear

plotly/matplotlylib/mplexporter/tests/test_basic.py

@@ -1,26 +1,34 @@
-from ..exporter import Exporter
-from ..renderers import FakeRenderer, FullFakeRenderer
+# TODO: matplotlib-build-wip
+from nose.plugins.attrib import attr
+from plotly.tools import _matplotlylib_imported
 
-import matplotlib
-matplotlib.use('Agg')
-import matplotlib.pyplot as plt
+if _matplotlylib_imported:
+    from ..exporter import Exporter
+    from ..renderers import FakeRenderer, FullFakeRenderer
 
-import numpy as np
-from numpy.testing import assert_warns
+    import matplotlib
+    matplotlib.use('Agg')
+    import matplotlib.pyplot as plt
 
+    import numpy as np
+    from numpy.testing import assert_warns
 
+
+@attr('matplotlib')
 def fake_renderer_output(fig, Renderer):
     renderer = Renderer()
     exporter = Exporter(renderer)
     exporter.run(fig)
     return renderer.output
 
 
+@attr('matplotlib')
 def _assert_output_equal(text1, text2):
     for line1, line2 in zip(text1.strip().split(), text2.strip().split()):
         assert line1 == line2
 
 
+@attr('matplotlib')
 def test_lines():
     fig, ax = plt.subplots()
     ax.plot(range(20), '-k')
@@ -44,6 +52,7 @@ def test_lines():
                          """)
 
 
+@attr('matplotlib')
 def test_markers():
     fig, ax = plt.subplots()
     ax.plot(range(2), 'ok')
@@ -68,6 +77,7 @@ def test_markers():
                          """)
 
 
+@attr('matplotlib')
 def test_path_collection():
     fig, ax = plt.subplots()
     ax.scatter(range(3), range(3))
@@ -93,6 +103,7 @@ def test_path_collection():
                          """)
 
 
+@attr('matplotlib')
 def test_text():
     fig, ax = plt.subplots()
     ax.set_xlabel("my x label")
@@ -113,6 +124,7 @@ def test_text():
                          """)
 
 
+@attr('matplotlib')
 def test_path():
     fig, ax = plt.subplots()
     ax.add_patch(plt.Circle((0, 0), 1))
@@ -129,6 +141,7 @@ def test_path():
                          """)
 
 
+@attr('matplotlib')
 def test_multiaxes():
     fig, ax = plt.subplots(2)
     ax[0].plot(range(4))
@@ -147,6 +160,7 @@ def test_multiaxes():
                          """)
 
 
+@attr('matplotlib')
 def test_image():
     np.random.seed(0)  # image size depends on the seed
     fig, ax = plt.subplots()
@@ -163,6 +177,7 @@ def test_image():
                          """)
 
 
+@attr('matplotlib')
 def test_legend():
     fig, ax = plt.subplots()
     ax.plot([1,2,3], label='label')
@@ -178,6 +193,8 @@ def test_legend():
                          closing figure
                          """)
 
+
+@attr('matplotlib')
 def test_legend_dots():
     fig, ax = plt.subplots()
     ax.plot([1,2,3], label='label')
@@ -200,8 +217,9 @@ def test_legend_dots():
                          closing figure
                          """)
 
+
+@attr('matplotlib')
 def test_blended():
     fig, ax = plt.subplots()
     ax.axvline(0)
     assert_warns(UserWarning, fake_renderer_output, fig, FakeRenderer)
-    

plotly/matplotlylib/mplexporter/tests/test_utils.py

@@ -1,8 +1,13 @@
-from numpy.testing import assert_allclose, assert_equal
-import matplotlib.pyplot as plt
-from .. import utils
+# TODO: matplotlib-build-wip
+from nose.plugins.attrib import attr
+from plotly.tools import _matplotlylib_imported
 
+if _matplotlylib_imported:
+    from numpy.testing import assert_allclose, assert_equal
+    import matplotlib.pyplot as plt
+    from .. import utils
 
+@attr('matplotlib')
 def test_path_data():
     circle = plt.Circle((0, 0), 1)
     vertices, codes = utils.SVG_path(circle.get_path())

plotly/tests/test_optional/optional_utils.py

@@ -1,15 +1,18 @@
 from __future__ import absolute_import
 
-import matplotlib
-# Force matplotlib to not use any Xwindows backend.
-matplotlib.use('Agg')
-
 import numpy as np
 
-from plotly.matplotlylib import Exporter, PlotlyRenderer
 from plotly.tests.utils import is_num_list
 from plotly.utils import get_by_path, node_generator
 
+# TODO: matplotlib-build-wip
+from plotly.tools import _matplotlylib_imported
+if _matplotlylib_imported:
+    import matplotlib
+    # Force matplotlib to not use any Xwindows backend.
+    matplotlib.use('Agg')
+    from plotly.matplotlylib import Exporter, PlotlyRenderer
+
 
 def run_fig(fig):
     renderer = PlotlyRenderer()

plotly/tests/test_optional/test_matplotlylib/test_annotations.py

@@ -1,15 +1,22 @@
 from __future__ import absolute_import
 
-import matplotlib
-# Force matplotlib to not use any Xwindows backend.
-matplotlib.use('Agg')
-import matplotlib.pyplot as plt
+from nose.plugins.attrib import attr
 
-from plotly.tests.utils import compare_dict
-from plotly.tests.test_optional.optional_utils import run_fig
-from plotly.tests.test_optional.test_matplotlylib.data.annotations import *
+# TODO: matplotlib-build-wip
+from plotly.tools import _matplotlylib_imported
 
+if _matplotlylib_imported:
+    import matplotlib
+    # Force matplotlib to not use any Xwindows backend.
+    matplotlib.use('Agg')
+    import matplotlib.pyplot as plt
 
+    from plotly.tests.utils import compare_dict
+    from plotly.tests.test_optional.optional_utils import run_fig
+    from plotly.tests.test_optional.test_matplotlylib.data.annotations import *
+
+
+@attr('matplotlib')
 def test_annotations():
     fig, ax = plt.subplots()
     ax.plot([1, 2, 3], 'b-')

plotly/tests/test_optional/test_matplotlylib/test_axis_scales.py

@@ -1,15 +1,22 @@
 from __future__ import absolute_import
 
-import matplotlib
-# Force matplotlib to not use any Xwindows backend.
-matplotlib.use('Agg')
-import matplotlib.pyplot as plt
+from nose.plugins.attrib import attr
 
 from plotly.tests.utils import compare_dict
 from plotly.tests.test_optional.optional_utils import run_fig
 from plotly.tests.test_optional.test_matplotlylib.data.axis_scales import *
 
+# TODO: matplotlib-build-wip
+from plotly.tools import _matplotlylib_imported
 
+if _matplotlylib_imported:
+    import matplotlib
+    # Force matplotlib to not use any Xwindows backend.
+    matplotlib.use('Agg')
+    import matplotlib.pyplot as plt
+
+
+@attr('matplotlib')
 def test_even_linear_scale():
     fig, ax = plt.subplots()
     x = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

plotly/tests/test_optional/test_matplotlylib/test_bars.py

@@ -1,15 +1,22 @@
 from __future__ import absolute_import
 
-import matplotlib
-# Force matplotlib to not use any Xwindows backend.
-matplotlib.use('Agg')
-import matplotlib.pyplot as plt
+from nose.plugins.attrib import attr
 
 from plotly.tests.utils import compare_dict
 from plotly.tests.test_optional.optional_utils import run_fig
 from plotly.tests.test_optional.test_matplotlylib.data.bars import *
 
+# TODO: matplotlib-build-wip
+from plotly.tools import _matplotlylib_imported
+if _matplotlylib_imported:
+    import matplotlib
 
+    # Force matplotlib to not use any Xwindows backend.
+    matplotlib.use('Agg')
+    import matplotlib.pyplot as plt
+
+
+@attr('matplotlib')
 def test_vertical_bar():
     fig, ax = plt.subplots()
     ax.bar(left=D['left'], height=D['height'])
@@ -23,6 +30,7 @@ def test_vertical_bar():
     assert equivalent, msg
 
 
+@attr('matplotlib')
 def test_horizontal_bar():
     fig, ax = plt.subplots()
     ax.barh(bottom=D['bottom'], width=D['width'])
@@ -36,6 +44,7 @@ def test_horizontal_bar():
     assert equivalent, msg
 
 
+@attr('matplotlib')
 def test_h_and_v_bars():
     fig, ax = plt.subplots()
     ax.bar(left=D['multi_left'], height=D['multi_height'],

plotly/tests/test_optional/test_matplotlylib/test_data.py

@@ -1,14 +1,21 @@
 from __future__ import absolute_import
 
-import matplotlib
-# Force matplotlib to not use any Xwindows backend.
-matplotlib.use('Agg')
-import matplotlib.pyplot as plt
+from nose.plugins.attrib import attr
 
 from plotly.tests.test_optional.optional_utils import run_fig
 from plotly.tests.test_optional.test_matplotlylib.data.data import *
 
+# TODO: matplotlib-build-wip
+from plotly.tools import _matplotlylib_imported
+if _matplotlylib_imported:
+    import matplotlib
 
+    # Force matplotlib to not use any Xwindows backend.
+    matplotlib.use('Agg')
+    import matplotlib.pyplot as plt
+
+
+@attr('matplotlib')
 def test_line_data():
     fig, ax = plt.subplots()
     ax.plot(D['x1'], D['y1'])
@@ -21,6 +28,7 @@ def test_line_data():
             renderer.plotly_fig['data'][0]['y']) + ' is not ' + str(D['y1'])
 
 
+@attr('matplotlib')
 def test_lines_data():
     fig, ax = plt.subplots()
     ax.plot(D['x1'], D['y1'])
@@ -40,6 +48,7 @@ def test_lines_data():
             renderer.plotly_fig['data'][0]['y']) + ' is not ' + str(D['y2'])
 
 
+@attr('matplotlib')
 def test_bar_data():
     fig, ax = plt.subplots()
     ax.bar(D['x1'], D['y1'])
@@ -49,6 +58,7 @@ def test_bar_data():
             renderer.plotly_fig['data'][0]['y']) + ' is not ' + str(D['y1'])
 
 
+@attr('matplotlib')
 def test_bars_data():
     fig, ax = plt.subplots()
     ax.bar(D['x1'], D['y1'], color='r')

plotly/tests/test_optional/test_matplotlylib/test_date_times.py

@@ -4,17 +4,25 @@
 import random
 from unittest import TestCase
 
-import matplotlib
-# Force matplotlib to not use any Xwindows backend.
-matplotlib.use('Agg')
-from matplotlib.dates import date2num
-import matplotlib.pyplot as plt
 import pandas as pd
+from nose.plugins.attrib import attr
 
 import plotly.tools as tls
 
+# TODO: matplotlib-build-wip
+from plotly.tools import _matplotlylib_imported
+if _matplotlylib_imported:
+    import matplotlib
 
+    # Force matplotlib to not use any Xwindows backend.
+    matplotlib.use('Agg')
+    from matplotlib.dates import date2num
+    import matplotlib.pyplot as plt
+
+
+@attr('matplotlib')
 class TestDateTimes(TestCase):
+
     def test_normal_mpl_dates(self):
         datetime_format = '%Y-%m-%d %H:%M:%S'
         y = [1, 2, 3, 4]

plotly/tests/test_optional/test_matplotlylib/test_lines.py

@@ -1,15 +1,22 @@
 from __future__ import absolute_import
 
-import matplotlib
-# Force matplotlib to not use any Xwindows backend.
-matplotlib.use('Agg')
-import matplotlib.pyplot as plt
+from nose.plugins.attrib import attr
 
 from plotly.tests.utils import compare_dict
 from plotly.tests.test_optional.optional_utils import run_fig
 from plotly.tests.test_optional.test_matplotlylib.data.lines import *
 
+# TODO: matplotlib-build-wip
+from plotly.tools import _matplotlylib_imported
+if _matplotlylib_imported:
+    import matplotlib
 
+    # Force matplotlib to not use any Xwindows backend.
+    matplotlib.use('Agg')
+    import matplotlib.pyplot as plt
+
+
+@attr('matplotlib')
 def test_simple_line():
     fig, ax = plt.subplots()
     ax.plot(D['x1'], D['y1'], label='simple')
@@ -22,6 +29,7 @@ def test_simple_line():
     assert equivalent, msg
 
 
+@attr('matplotlib')
 def test_complicated_line():
     fig, ax = plt.subplots()
     ax.plot(D['x1'], D['y1'], 'ro', markersize=10, alpha=.5, label='one')