Merge pull request #12651 from jklymak/fix-ignore-non-finite-bbox

anntzer · web-flow · commit 41f72fbe4491 · 2018-10-29T20:35:19.000+01:00
FIX: ignore non-finite bbox
diff --git a/lib/matplotlib/axes/_base.py b/lib/matplotlib/axes/_base.py
@@ -4207,7 +4207,9 @@ def get_tightbbox(self, renderer, call_axes_locator=True,
 
         for a in bbox_artists:
             bbox = a.get_tightbbox(renderer)
-            if bbox is not None and (bbox.width != 0 or bbox.height != 0):
+            if (bbox is not None and
+                    (bbox.width != 0 or bbox.height != 0) and
+                    np.isfinite(bbox.width) and np.isfinite(bbox.height)):
                 bb.append(bbox)
 
         _bbox = mtransforms.Bbox.union(
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -5849,3 +5849,10 @@ class DummySubplot(matplotlib.axes.SubplotBase, Dummy):
     FactoryDummySubplot = matplotlib.axes.subplot_class_factory(Dummy)
 
     assert DummySubplot is FactoryDummySubplot
+
+
+def test_gettightbbox_ignoreNaN():
+    fig, ax = plt.subplots()
+    t = ax.text(np.NaN, 1, 'Boo')
+    renderer = fig.canvas.get_renderer()
+    np.testing.assert_allclose(ax.get_tightbbox(renderer).width, 532.444444)
