Test for set_array in Collection

aitikgupta · aitikgupta · commit e7efa101e226 · 2020-11-02T10:14:03.000+05:30
diff --git a/lib/matplotlib/tests/test_collections.py b/lib/matplotlib/tests/test_collections.py
@@ -649,6 +649,15 @@ def test_collection_set_verts_array():
         assert np.array_equal(ap._codes, atp._codes)
 
 
+def test_collection_set_array():
+    xs = np.arange(10)
+    ymin = 0
+    ymax = 1
+    vals = [*range(10)]
+
+    plt.vlines(xs, ymin, ymax, array=vals)
+
+
 def test_blended_collection_autolim():
     a = [1, 2, 4]
     height = .2
