TST: add a test

jklymak · jklymak · commit 2df24aab8130 · 2020-05-26T08:36:57.000-07:00
diff --git a/lib/matplotlib/tests/test_figure.py b/lib/matplotlib/tests/test_figure.py
@@ -566,3 +566,18 @@ def test_add_subplot_twotuple():
     assert ax4.get_subplotspec().colspan == range(0, 2)
     with pytest.raises(IndexError):
         fig.add_subplot(3, 2, (6, 3))
+
+
+def test_reused_gridspec():
+    """Test that these all use the same gridspec"""
+    fig = plt.figure()
+    ax1 = fig.add_subplot(3, 2, (3, 5))
+    ax2 = fig.add_subplot(3, 2, 4)
+    ax3 = plt.subplot2grid((3, 2), (2, 1), colspan=2, fig=fig)
+
+    gs1 = ax1.get_subplotspec().get_gridspec()
+    gs2 = ax2.get_subplotspec().get_gridspec()
+    gs3 = ax3.get_subplotspec().get_gridspec()
+
+    assert gs1 == gs2
+    assert gs1 == gs3
