autoscale arrow

choyiny · choyiny · commit 90794e465151 · 2019-10-09T14:48:48.000-04:00
diff --git a/doc/api/next_api_changes/behaviour.rst b/doc/api/next_api_changes/behaviour.rst
@@ -17,3 +17,7 @@ Previously, it did not necessarily have such an attribute.  A check for
 ``hasattr(figure.canvas, "manager")`` should now be replaced by
 ``figure.canvas.manager is not None`` (or ``getattr(figure.canvas, "manager", None) is not None``
 for back-compatibility).
+
+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
@@ -4938,7 +4938,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
@@ -382,6 +382,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()
