matplotlib
diff --git a/‎lib/matplotlib/tests/baseline_images/test_collections/cap_and_joinstyle.png‎
29.7 KB b/‎lib/matplotlib/tests/baseline_images/test_collections/cap_and_joinstyle.png‎
29.7 KB
diff --git a/‎lib/matplotlib/tests/test_collections.py‎
Lines changed: 23 additions & 1 deletion b/‎lib/matplotlib/tests/test_collections.py‎
Lines changed: 23 additions & 1 deletion
@@ -14,7 +14,7 @@
 import matplotlib.pyplot as plt
 import matplotlib.collections as mcollections
 import matplotlib.transforms as mtransforms
-from matplotlib.collections import Collection, EventCollection
+from matplotlib.collections import Collection, LineCollection, EventCollection
 from matplotlib.testing.decorators import image_comparison
 
 
@@ -642,6 +642,28 @@ def test_joinstyle():
     assert_equal(col.get_joinstyle(), 'miter')
 
 
+@image_comparison(baseline_images=['cap_and_joinstyle'],
+                  extensions=['png'])
+def test_cap_and_joinstyle_image():
+    fig = plt.figure()
+    ax = fig.add_subplot(1, 1, 1)
+    ax.set_xlim([-0.5, 1.5])
+    ax.set_ylim([-0.5, 2.5])
+
+    x = np.array([0.0, 1.0, 0.5])
+    ys = np.array([[0.0], [0.5], [1.0]]) + np.array([[0.0, 0.0, 1.0]])
+
+    segs = np.zeros((3, 3, 2))
+    segs[:, :, 0] = x
+    segs[:, :, 1] = ys
+    line_segments = LineCollection(segs, linewidth=[10, 15, 20])
+    line_segments.set_capstyle("round")
+    line_segments.set_joinstyle("miter")
+
+    ax.add_collection(line_segments)
+    ax.set_title('Line collection with customized caps and joinstyle')
+
+
 @image_comparison(baseline_images=['scatter_post_alpha'],
                   extensions=['png'], remove_text=True,
                   style='default')