Fixes a bug that caused the use of _as_mpl_axes to crash if one of the returned axes arguments defined __getitem__ but was not iterable.

astrofrog · astrofrog · commit dab0fcf35cbc · 2014-11-26T21:45:50.000+01:00
diff --git a/lib/matplotlib/figure.py b/lib/matplotlib/figure.py
@@ -786,7 +786,13 @@ def fixitems(items):
             ret = []
             for k, v in items:
                 if iterable(v):
-                    v = tuple(v)
+                    # some objects can define __getitem__ without being 
+                    # iterable and in those cases the conversion to tuples 
+                    # will fail.
+                    try:
+                        v = tuple(v)
+                    except:
+                        pass
                 ret.append((k, v))
             return tuple(ret)
 
diff --git a/lib/matplotlib/tests/test_figure.py b/lib/matplotlib/tests/test_figure.py
@@ -6,6 +6,7 @@
 
 from nose.tools import assert_equal, assert_true, assert_raises
 from matplotlib.testing.decorators import image_comparison, cleanup
+from matplotlib.axes import Axes
 import matplotlib.pyplot as plt
 
 
@@ -110,6 +111,36 @@ def test_too_many_figures():
     assert len(w) == 1
 
 
+def test_iterability_axes_argument():
+
+    # This is a regression test for matplotlib/matplotlib#3196. If one of the
+    # arguments returned by _as_mpl_axes defines __getitem__ but is not
+    # iterable, this would raise an execption. This is because we check
+    # whether the arguments are iterable, and if so we try and convert them
+    # to a tuple. However, the ``iterable`` function returns True if
+    # __getitem__ is present, but some classes can define __getitem__ without
+    # being iterable. The tuple conversion is now done in a try...except in
+    # case it fails.
+
+    class MyAxes(Axes):
+        def __init__(self, *args, **kwargs):
+            kwargs.pop('myclass', None)
+            return Axes.__init__(self, *args, **kwargs)
+
+    class MyClass(object):
+
+        def __getitem__(self, item):
+            if item != 'a':
+                raise ValueError("item should be a")
+
+        def _as_mpl_axes(self):
+            return MyAxes, {'myclass': self}
+
+    fig = plt.figure()
+    ax = fig.add_subplot(1, 1, 1, projection=MyClass())
+    plt.close(fig)
+
+
 if __name__ == "__main__":
     import nose
     nose.runmodule(argv=['-s', '--with-doctest'], exit=False)
