FIX: make sure that actual bboxes also work

jklymak · jklymak · commit d973d9cd1d7b · 2021-01-22T17:13:05.000-08:00
diff --git a/lib/matplotlib/backend_bases.py b/lib/matplotlib/backend_bases.py
@@ -2196,24 +2196,27 @@ def print_figure(
             self.figure.set_facecolor(facecolor)
             self.figure.set_edgecolor(edgecolor)
 
-            cl_state = self.figure.get_constrained_layout()
-
             if bbox_inches is None:
                 bbox_inches = rcParams['savefig.bbox']
+
+            if (self.figure.get_constrained_layout() or
+                    bbox_inches == "tight"):
+                # we need to trigger a draw before printing to make sure
+                # CL works.  "tight" also needs a draw to get the right
+                # locations:
+                renderer = _get_renderer(
+                    self.figure,
+                    functools.partial(
+                        print_method, orientation=orientation)
+                )
+                ctx = (renderer._draw_disabled()
+                       if hasattr(renderer, '_draw_disabled')
+                       else suppress())
+                with ctx:
+                    self.figure.draw(renderer)
+
             if bbox_inches:
                 if bbox_inches == "tight":
-                    renderer = _get_renderer(
-                        self.figure,
-                        functools.partial(
-                            print_method, orientation=orientation)
-                    )
-                    ctx = (renderer._draw_disabled()
-                           if hasattr(renderer, '_draw_disabled')
-                           else suppress())
-                    with ctx:
-                        self.figure.draw(renderer)
-                        self.figure.set_constrained_layout(False)
-
                     bbox_inches = self.figure.get_tightbbox(
                         renderer, bbox_extra_artists=bbox_extra_artists)
                     if pad_inches is None:
@@ -2229,6 +2232,9 @@ def print_figure(
                 _bbox_inches_restore = None
 
             try:
+                # we have already done CL above, so turn it off:
+                cl_state = self.figure.get_constrained_layout()
+                self.figure.set_constrained_layout(False)
                 result = print_method(
                     filename,
                     dpi=dpi,
@@ -2244,6 +2250,7 @@ def print_figure(
                 self.figure.set_facecolor(origfacecolor)
                 self.figure.set_edgecolor(origedgecolor)
                 self.figure.set_canvas(self)
+            # reset to cached state
             self.figure.set_constrained_layout(cl_state)
             return result
 
diff --git a/lib/matplotlib/tests/baseline_images/test_constrainedlayout/test_bbox.png b/lib/matplotlib/tests/baseline_images/test_constrainedlayout/test_bbox.png
diff --git a/lib/matplotlib/tests/test_constrainedlayout.py b/lib/matplotlib/tests/test_constrainedlayout.py
@@ -4,9 +4,11 @@
 from matplotlib.testing.decorators import image_comparison
 import matplotlib.pyplot as plt
 import matplotlib.gridspec as gridspec
+import matplotlib.transforms as mtransforms
 from matplotlib import ticker, rcParams
 
 
+
 def example_plot(ax, fontsize=12, nodec=False):
     ax.plot([1, 2])
     ax.locator_params(nbins=3)
@@ -511,3 +513,12 @@ def test_manually_set_position():
 def test_bboxtight():
     fig, ax = plt.subplots(constrained_layout=True)
     ax.set_aspect(1.)
+
+
+@image_comparison(['test_bbox.png'],
+                  remove_text=True, style='mpl20',
+                  savefig_kwarg={'bbox_inches':
+                                 mtransforms.Bbox([[0.5, 0], [2.5, 2]])})
+def test_bbox():
+    fig, ax = plt.subplots(constrained_layout=True)
+    ax.set_aspect(1.)
