TST: test hidden axes

jklymak · jklymak · commit c8ec3c67fe51 · 2019-08-08T14:18:59.000-07:00
Update lib/matplotlib/tests/test_constrainedlayout.py

Co-Authored-By: David Stansby &lt;dstansby@gmail.com&gt;

FLAKE8
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)
