Added the old test back in to prevent error handling change. Also unit test!

WeatherGod · WeatherGod · commit 8a722c17ad6a · 2015-11-12T16:35:56.000-05:00
diff --git a/lib/matplotlib/figure.py b/lib/matplotlib/figure.py
@@ -1758,7 +1758,7 @@ def figaspect(arg):
     Thanks to Fernando Perez for this function
     """
 
-    isarray = not np.isscalar(arg)
+    isarray = hasattr(arg, 'shape') and not np.isscalar(arg)
 
     # min/max sizes to respect when autoscaling.  If John likes the idea, they
     # could become rc parameters, for now they're hardwired.
diff --git a/lib/matplotlib/tests/test_figure.py b/lib/matplotlib/tests/test_figure.py
@@ -8,7 +8,7 @@
 from matplotlib.testing.decorators import image_comparison, cleanup
 from matplotlib.axes import Axes
 import matplotlib.pyplot as plt
-
+import numpy as np
 
 @cleanup
 def test_figure_label():
@@ -191,6 +191,17 @@ def test_axes_remove():
     assert_equal(len(fig.axes), 3)
 
 
+def test_figaspect():
+    w, h = plt.figaspect(np.float64(2) / np.float64(1))
+    assert h / w == 2
+    w, h = plt.figaspect(2)
+    assert h / w == 2
+    w, h = plt.figaspect(np.zeros((1, 2)))
+    assert h / w == 0.5
+    w, h = plt.figaspect(np.zeros((2, 2)))
+    assert h / w == 1
+
+
 if __name__ == "__main__":
     import nose
     nose.runmodule(argv=['-s', '--with-doctest'], exit=False)
