Test pyplot.axes(ax) with ax from other figure.

pavoljuhas · pavoljuhas · commit 712116ba6ccd · 2017-02-08T11:43:09.000-05:00
Ensure `ax` becomes the current axis and its parent the current figure.
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -1476,6 +1476,15 @@ def _as_mpl_axes(self):
         'Expected a PolarAxesSubplot, got %s' % type(ax)
     plt.close()
 
+    # test focusing of Axes in other Figure
+    fig1, ax1 = subplots()
+    fig2, ax2 = subplots()
+    assert ax1 is plt.axes(ax1)
+    assert ax1 is plt.gca()
+    assert fig1 is plt.gcf()
+    plt.close(fig1)
+    plt.close(fig2)
+
 
 @image_comparison(baseline_images=['log_scales'])
 def test_log_scales():
