TST: adjusting existing axes with collage

tacaswell · tacaswell · commit f25f1f7e6100 · 2020-03-01T23:38:09.000-05:00
diff --git a/lib/matplotlib/tests/test_figure.py b/lib/matplotlib/tests/test_figure.py
@@ -595,3 +595,20 @@ def test_empty(self, fig_test, fig_ref, x, empty_sentinel):
 
         axB = fig_ref.add_subplot(gs[1, 1])
         axB.set_title(labels[1])
+
+    @check_figures_equal()
+    def test_adjust_existing(self, fig_test, fig_ref):
+        x = [["A", "B"], ["C", "B"]]
+
+        grid_axes = fig_ref.build_grid(x)
+
+        for k, ax in grid_axes.items():
+            ax.set_title(k)
+
+        (axA, axB, axC) = fig_test.subplots(3)
+        for ax, title in zip((axA, axB, axC), 'ABC'):
+            ax.set_title(title)
+
+        y = [[axA, axB], [axC, axB]]
+
+        grid_axes_ts = fig_test.build_grid(y)
