TST: add a test

jklymak · jklymak · commit 3d9fbcc750de · 2020-05-06T19:52:39.000-07:00
diff --git a/lib/matplotlib/tests/test_figure.py b/lib/matplotlib/tests/test_figure.py
@@ -546,3 +546,19 @@ def test_add_subplot_twotuple():
     ax4 = fig.add_subplot(3, 2, (3, 6))
     assert ax4.get_subplotspec().rowspan == range(1, 3)
     assert ax4.get_subplotspec().colspan == range(0, 2)
+
+
+def test_reuse_gridspec():
+    fig = plt.figure()
+    ax = [None] * 9
+    for i in range(9):
+        ax[i] = fig.add_subplot(3, 3, i + 1)
+    for i in range(1, 9):
+        assert (ax[i].get_subplotspec().get_gridspec() ==
+                ax[0].get_subplotspec().get_gridspec())
+
+    fig = plt.figure()
+    ax1 = fig.add_subplot(3, 3, (1, 7))
+    ax2 = fig.add_subplot(3, 3, (6, 9))
+    assert (ax1.get_subplotspec().get_gridspec() ==
+            ax2.get_subplotspec().get_gridspec())
