Merge pull request #15392 from choyiny/arrow-autoscale

QuLogic · web-flow · commit defce2e1225f · 2020-03-24T22:06:48.000-04:00
Autoscale for ax.arrow()
diff --git a/doc/api/next_api_changes/behaviour.rst b/doc/api/next_api_changes/behaviour.rst
@@ -129,3 +129,7 @@ Shortcuts for closing all figures now also work for the classic toolbar.
 There is no default shortcut any more because unintentionally closing all figures by a key press
 might happen too easily. You can configure the shortcut yourself
 using :rc:`keymap.quit_all`.
+
+Autoscale for arrow
+~~~~~~~~~~~~~~~~~~~
+Calling ax.arrow() will now autoscale the axes.
diff --git a/lib/matplotlib/axes/_axes.py b/lib/matplotlib/axes/_axes.py
@@ -4959,7 +4959,8 @@ def arrow(self, x, y, dx, dy, **kwargs):
         dy = self.convert_yunits(dy)
 
         a = mpatches.FancyArrow(x, y, dx, dy, **kwargs)
-        self.add_artist(a)
+        self.add_patch(a)
+        self._request_autoscale_view()
         return a
 
     @docstring.copy(mquiver.QuiverKey.__init__)
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -468,6 +468,13 @@ def test_arrow_empty():
     ax.arrow(0, 0, 0, 0, head_length=0)
 
 
+def test_arrow_in_view():
+    _, ax = plt.subplots()
+    ax.arrow(1, 1, 1, 1)
+    assert ax.get_xlim() == (0.8, 2.2)
+    assert ax.get_ylim() == (0.8, 2.2)
+
+
 def test_annotate_default_arrow():
     # Check that we can make an annotation arrow with only default properties.
     fig, ax = plt.subplots()
