datetime vs number, number vs datetime, datetime vs datetime

Author: danielcobej

lib/matplotlib/tests/test_datetime.py

@@ -35,24 +35,39 @@ def test_axhline(self):
     @mpl.style.context("default")
     def test_axhspan(self):
         mpl.rcParams["date.converter"] = 'concise'
-        np.random.seed(19680801)
 
         start_date = datetime.datetime(2023, 1, 1)
-        time_delta = datetime.timedelta(days=1)
-
-        fig, (ax1, ax2) = plt.subplots(2, 1, constrained_layout=True, figsize=(10, 8))
-
-        ax1.set_ylim(start_date, start_date + 29*time_delta)
-        for i in range(np.random.randint(1, 5)):
-            ymin = start_date + np.random.randint(0, 30) * time_delta
-            ymax = ymin + np.random.randint(1, 3) * time_delta
-            ax1.axhspan(ymin=ymin, ymax=ymax, facecolor='green', alpha=0.5)
-
-        ax2.set_ylim(start_date, start_date + 29*time_delta)
-        for i in range(0, 30, 2):
-            ymin = start_date + i * time_delta
-            ymax = ymin + time_delta
+        dates = [start_date + datetime.timedelta(days=i) for i in range(31)]
+        numbers = list(range(1, 32))
+
+        fig, (ax1, ax2, ax3) = plt.subplots(3, 1,
+                                            constrained_layout=True,
+                                            figsize=(10, 12))
+
+        ax1.plot(dates, numbers, marker='o', color='blue')
+        for i in range(0, 31, 2):
+            ax1.axhspan(ymin=i+1, ymax=i+2, facecolor='green', alpha=0.5)
+        ax1.set_title('Datetime vs. Number')
+        ax1.set_xlabel('Date')
+        ax1.set_ylabel('Number')
+
+        ax2.plot(numbers, dates, marker='o', color='blue')
+        for i in range(0, 31, 2):
+            ymin = start_date + datetime.timedelta(days=i)
+            ymax = ymin + datetime.timedelta(days=1)
             ax2.axhspan(ymin=ymin, ymax=ymax, facecolor='green', alpha=0.5)
+        ax2.set_title('Number vs. Datetime')
+        ax2.set_xlabel('Number')
+        ax2.set_ylabel('Date')
+
+        ax3.plot(dates, dates, marker='o', color='blue')
+        for i in range(0, 31, 2):
+            ymin = start_date + datetime.timedelta(days=i)
+            ymax = ymin + datetime.timedelta(days=1)
+            ax3.axhspan(ymin=ymin, ymax=ymax, facecolor='green', alpha=0.5)
+        ax3.set_title('Datetime vs. Datetime')
+        ax3.set_xlabel('Date')
+        ax3.set_ylabel('Date')
 
     @pytest.mark.xfail(reason="Test for axline not written yet")
     @mpl.style.context("default")