matplotlib
diff --git a/‎lib/matplotlib/tests/baseline_images/test_axes/test_eventplot_colors.png
-5.4 KB b/‎lib/matplotlib/tests/baseline_images/test_axes/test_eventplot_colors.png
-5.4 KB
diff --git a/‎lib/matplotlib/tests/test_axes.py
Lines changed: 21 additions & 17 deletions b/‎lib/matplotlib/tests/test_axes.py
Lines changed: 21 additions & 17 deletions
@@ -2909,28 +2909,32 @@ def test_eventplot_defaults():
     colls = axobj.eventplot(data)
 
 
-@image_comparison(baseline_images=['test_eventplot_colors'],
-                  extensions=['png'], remove_text=True)
-def test_eventplot_colors():
-    '''
-    Test that eventplot produces the correct output in several cases of
-    the *colors* parameter. Inspired by the issue #8193.
+@pytest.mark.parametrize(('colors'), [
+    ('0.5',), # string color with multiple characters: not OK before #8193 fix
+    ('tab:orange', 'tab:pink', 'tab:cyan', 'bLacK'),  # case-insensitive
+    ('red', (0, 1, 0), None, (1, 0, 1, 0.5)),  # a tricky case mixing types
+    ('rgbk',)  # len('rgbk') == len(data) and each character is a valid color
+])
+def test_eventplot_colors(colors):
+    '''Test the *colors* parameter of eventplot. Inspired by the issue #8193.
     '''
     data = [[i] for i in range(4)]  # 4 successive events of different nature
 
-    fig, axs = plt.subplots(nrows=2, ncols=2)
+    # Build the list of the expected colors
+    expected = [c if c is not None else 'C0' for c in colors]
+    # Convert the list into an array of RGBA values
+    # NB: ['rgbk'] is not a valid argument for to_rgba_array, while 'rgbk' is.
+    if len(expected) == 1:
+        expected = expected[0]
+    expected = np.broadcast_to(mcolors.to_rgba_array(expected), (len(data), 4))
 
-    axs[0, 0].eventplot(data, colors='0.5')  # not passing before #8193 fix
-    axs[0, 1].eventplot(data, colors=['tab:orange', 'tab:pink', 'tab:cyan',
-                                      'bLacK'])  # should be case-insensitive
-    # A tricky case that mixes named, RGB, None and RGBA values
-    axs[1, 0].eventplot(data, colors=['red', (0, 1, 0), None, (1, 0, 1, 0.5)])
-    # Mostly a corner case, but it was passing even before the fix for #8193:
-    # len('rgbk') == len(data) and each single character is a valid color
-    axs[1, 1].eventplot(data, colors='rgbk')
+    fig, ax = plt.subplots()
+    if len(colors) == 1:  # tuple with a single string (like '0.5' or 'rgbk')
+        colors = colors[0]
+    collections = ax.eventplot(data, colors=colors)
 
-    for ax in axs.flat:
-        ax.set_xlim(-0.1, 3.1)  # avoid having some events on the axes edges
+    for coll, color in zip(collections, expected):
+        assert_allclose(coll.get_color(), color)
 
 
 @image_comparison(baseline_images=['test_eventplot_problem_kwargs'],