TST: test hidden axes

jklymak · dstansby · jklymak · commit 18538e23de9a · 2019-08-06T11:32:27.000-07:00
Update lib/matplotlib/tests/test_constrainedlayout.py

Co-Authored-By: David Stansby &lt;dstansby@gmail.com&gt;
diff --git a/lib/matplotlib/_constrained_layout.py b/lib/matplotlib/_constrained_layout.py
@@ -276,7 +276,7 @@ def _make_layout_margins(ax, renderer, h_pad, w_pad):
         bbox = pos
     else:
         bbox = invTransFig(tightbbox)
-
+    
     # this can go wrong:
     if not (np.isfinite(bbox.width) and np.isfinite(bbox.height)):
         # just abort, this is likely a bad set of co-ordinates that
diff --git a/lib/matplotlib/tests/test_constrainedlayout.py b/lib/matplotlib/tests/test_constrainedlayout.py
@@ -387,3 +387,16 @@ def test_colorbar_location():
     fig.colorbar(pcm, ax=axs[-2, 3:], shrink=0.5, location='top')
     fig.colorbar(pcm, ax=axs[0, 0], shrink=0.5, location='left')
     fig.colorbar(pcm, ax=axs[1:3, 2], shrink=0.5, location='right')
+
+
+def test_hidden_axes():
+    # test that if we make an axes not visible that constrained_layout
+    # still works.  Note the axes still takes space in the layout
+    # (as does a gridspec slot that is empty)
+    fig, axs = plt.subplots(2, 2, constrained_layout=True)
+    axs[0, 1].set_visible(False)
+    fig.canvas.draw()
+    extents1 = np.copy(axs[0, 0].get_position().extents)
+
+    np.testing.assert_allclose(extents1,
+        [0.045552, 0.548288, 0.47319, 0.982638], rtol=1e-5)
