TST: add test and fix issue with keyword

jklymak · jklymak · commit a41d69b1d332 · 2021-10-29T13:26:37.000+02:00
diff --git a/lib/matplotlib/pyplot.py b/lib/matplotlib/pyplot.py
@@ -1059,8 +1059,8 @@ def axes(arg=None, **kwargs):
         plt.axes((left, bottom, width, height), facecolor='w')
     """
     fig = gcf()
+    pos = kwargs.pop('position', None)
     if arg is None:
-        pos = kwargs.pop('position', None)
         if pos is None:
             return fig.add_subplot(**kwargs)
         else:
diff --git a/lib/matplotlib/tests/test_pyplot.py b/lib/matplotlib/tests/test_pyplot.py
@@ -1,4 +1,5 @@
 import difflib
+import numpy as np
 import subprocess
 import sys
 from pathlib import Path
@@ -320,3 +321,15 @@ def test_polar_second_call():
     ln2, = plt.polar(1.57, .5, 'bo')
     assert isinstance(ln2, mpl.lines.Line2D)
     assert ln1.axes is ln2.axes
+
+
+def test_fallback_position():
+    # check that position kwarg works if rect not supplied
+    axref = plt.axes([0.2, 0.2, 0.5, 0.5])
+    axtest = plt.axes(position=[0.2, 0.2, 0.5, 0.5])
+    np.testing.assert_allclose(axtest.bbox.get_points(), axref.bbox.get_points())
+
+    # check that position kwarg ignored if rect is supplied
+    axref = plt.axes([0.2, 0.2, 0.5, 0.5])
+    axtest = plt.axes([0.2, 0.2, 0.5, 0.5], position=[0.1, 0.1, 0.8, 0.8])
+    np.testing.assert_allclose(axtest.bbox.get_points(), axref.bbox.get_points())
