matplotlib
diff --git a/‎lib/matplotlib/tests/baseline_images/test_axes/arrow_simple.png
14.7 KB b/‎lib/matplotlib/tests/baseline_images/test_axes/arrow_simple.png
14.7 KB
diff --git a/‎lib/matplotlib/tests/test_axes.py
Lines changed: 29 additions & 1 deletion b/‎lib/matplotlib/tests/test_axes.py
Lines changed: 29 additions & 1 deletion
@@ -3,7 +3,7 @@
 
 import six
 from six.moves import xrange
-from itertools import chain
+from itertools import chain, product
 from distutils.version import LooseVersion
 import io
 
@@ -263,6 +263,34 @@ def test_basic_annotate():
                 xytext=(3, 3), textcoords='offset points')
 
 
+@image_comparison(baseline_images=['arrow_simple'],
+                  extensions=['png'], remove_text=True)
+def test_arrow_simple():
+    # Simple image test for ax.arrow
+    # kwargs that take discrete values
+    length_includes_head = (True, False)
+    shape = ('full', 'left', 'right')
+    head_starts_at_zero = (True, False)
+    # Create outer product of values
+    kwargs = list(product(length_includes_head, shape, head_starts_at_zero))
+
+    fig, axs = plt.subplots(3, 4)
+    for i, (ax, kwarg) in enumerate(zip(axs.flatten(), kwargs)):
+        ax.set_xlim(-2, 2)
+        ax.set_ylim(-2, 2)
+        # Unpack kwargs
+        (length_includes_head, shape, head_starts_at_zero) = kwarg
+        theta = 2 * np.pi * i / 12
+        # Draw arrow
+        ax.arrow(0, 0, np.sin(theta), np.cos(theta),
+                 width=theta/100,
+                 length_includes_head=length_includes_head,
+                 shape=shape,
+                 head_starts_at_zero=head_starts_at_zero,
+                 head_width=theta / 10,
+                 head_length=theta / 10)
+
+
 def test_annotate_default_arrow():
     # Check that we can make an annotation arrow with only default properties.
     fig, ax = plt.subplots()