boost supported mpl version again

Author: keflavich

docs/visualization.rst

@@ -34,7 +34,7 @@ to plot an image slice.
 .. warning:: 
 
    There are known incompatibilities with the above plotting approach:
-   matplotlib versions ``<2`` will crash, and you will have to clear the plot
+   matplotlib versions ``<2.1`` will crash, and you will have to clear the plot
    window to reset it.
 
 

spectral_cube/tests/test_visualization.py

@@ -1,6 +1,7 @@
 from __future__ import print_function, absolute_import, division
 
 import pytest
+from distutils.version import StrictVersion
 
 from .test_spectral_cube import cube_and_raw
 
@@ -22,8 +23,9 @@ def test_proj_imshow():
     plt = pytest.importorskip('matplotlib.pyplot')
     cube, data = cube_and_raw('vda_Jybeam_lower.fits')
     mom0 = cube.moment0()
-    if int(plt.matplotlib.__version__[0]) < 2:
+    if StrictVersion(plt.matplotlib.__version__) < StrictVersion('2.1'):
         # imshow is now only compatible with more recent versions of matplotlib
+        # (apparently 2.0.2 was still incompatible)
         plt.imshow(mom0.value)
     else:
         plt.imshow(mom0)