Add test_eventplot to test_datetime.py

Shriya Kalakata · Shriya Kalakata · commit 4e45a67c6a13 · 2023-12-07T17:49:30.000-05:00
diff --git a/lib/matplotlib/tests/test_datetime.py b/lib/matplotlib/tests/test_datetime.py
@@ -260,11 +260,36 @@ def test_errorbar(self):
                      uplims=True, xuplims=True,
                      label='Data')
 
-    @pytest.mark.xfail(reason="Test for eventplot not written yet")
     @mpl.style.context("default")
     def test_eventplot(self):
-        fig, ax = plt.subplots()
-        ax.eventplot(...)
+        mpl.rcParams["date.converter"] = "concise"
+
+        fig, (ax1, ax2) = plt.subplots(2, 1, layout="constrained")
+
+        x_dates1 = np.array([
+            datetime.datetime(2020, 6, 30),
+            datetime.datetime(2020, 7, 22),
+            datetime.datetime(2020, 8, 3),
+            datetime.datetime(2020, 9, 14),
+        ], dtype=np.datetime64)
+
+        ax1.eventplot(x_dates1)
+
+        x_dates2 = np.array([
+            [datetime.datetime(2020, 6, 30), datetime.datetime(2020, 7, 22),
+             datetime.datetime(2020, 8, 3), datetime.datetime(2020, 9, 14)],
+            [datetime.datetime(2020, 7, 18), datetime.datetime(2020, 7, 21),
+             datetime.datetime(2020, 8, 3), datetime.datetime(2020, 10, 14)]
+        ], dtype=np.datetime64)
+
+        colors = ['C{}'.format(i) for i in range(2)]
+        lineoffsets = np.array([1, 6])
+        linelengths = [5, 2]
+
+        ax2.eventplot(x_dates2,
+                      colors=colors,
+                      lineoffsets=lineoffsets,
+                      linelengths=linelengths)
 
     @pytest.mark.xfail(reason="Test for fill not written yet")
     @mpl.style.context("default")
