Modified test to check for input with varying lengths of data

Shriya Kalakata · Shriya Kalakata · commit a676dce42268 · 2023-12-07T23:19:45.000-05:00
diff --git a/lib/matplotlib/tests/test_datetime.py b/lib/matplotlib/tests/test_datetime.py
@@ -266,30 +266,31 @@ def test_eventplot(self):
 
         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)
+        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)
+        start_date = datetime.datetime(2020, 7, 1)
+        end_date = datetime.datetime(2020, 10, 15)
+        date_range = end_date - start_date
+
+        dates1 = start_date + np.random.rand(30) * date_range
+        dates2 = start_date + np.random.rand(10) * date_range
+        dates3 = start_date + np.random.rand(50) * date_range
+
+        colors1 = ['C1', 'C2', 'C3']
+        lineoffsets1 = np.array([1, 6, 8])
+        linelengths1 = [5, 2, 3]
+
+        ax2.eventplot([dates1, dates2, dates3],
+                      colors=colors1,
+                      lineoffsets=lineoffsets1,
+                      linelengths=linelengths1)
 
     @pytest.mark.xfail(reason="Test for fill not written yet")
     @mpl.style.context("default")
