matplotlib · Kojoley · Oct 28, 2017 · Oct 22, 2017 · Oct 22, 2017 · tacaswell
diff --git a/lib/matplotlib/tests/baseline_images/test_arrow_patches/arrow_styles.png b/lib/matplotlib/tests/baseline_images/test_arrow_patches/arrow_styles.png
diff --git a/lib/matplotlib/tests/baseline_images/test_patches/connection_patch.png b/lib/matplotlib/tests/baseline_images/test_patches/connection_patch.png
diff --git a/lib/matplotlib/tests/test_arrow_patches.py b/lib/matplotlib/tests/test_arrow_patches.py
@@ -119,3 +119,20 @@ def test_fancyarrow_dash():
                         color='k')
    ax.add_patch(e)
    ax.add_patch(e2)
+
+
+@image_comparison(baseline_images=['arrow_styles'], extensions=['png'],
+                  style='mpl20', remove_text=True)
+def test_arrow_styles():
+    styles = mpatches.ArrowStyle.get_styles()
+
+    n = len(styles)
+    fig, ax = plt.subplots(figsize=(6, 10))
+    ax.set_xlim(0, 1)
+    ax.set_ylim(-1, n)
+
+    for i, stylename in enumerate(sorted(styles)):
+        patch = mpatches.FancyArrowPatch((0.1, i), (0.8, i),
+                                         arrowstyle=stylename,
+                                         mutation_scale=25)
+        ax.add_patch(patch)
diff --git a/lib/matplotlib/tests/test_patches.py b/lib/matplotlib/tests/test_patches.py
@@ -335,7 +335,7 @@ def test_multi_color_hatch():

    for i in range(5):
        with mstyle.context({'hatch.color': 'C{}'.format(i)}):
-            r = Rectangle((i-.8/2, 5), .8, 1, hatch='//', fc='none')
+            r = Rectangle((i - .8 / 2, 5), .8, 1, hatch='//', fc='none')
        ax.add_patch(r)


@@ -350,3 +350,15 @@ def test_units_rectangle():
    ax.add_patch(p)
    ax.set_xlim([4*U.km, 7*U.km])
    ax.set_ylim([5*U.km, 9*U.km])
+
+
+@image_comparison(baseline_images=['connection_patch'], extensions=['png'],
+                  style='mpl20', remove_text=True)
+def test_connection_patch():
+    fig, (ax1, ax2) = plt.subplots(1, 2)
+
+    con = mpatches.ConnectionPatch(xyA=(0.1, 0.1), xyB=(0.9, 0.9),
+                                   coordsA='data', coordsB='data',
+                                   axesA=ax2, axesB=ax1,
+                                   arrowstyle="->")
+    ax2.add_artist(con)