Make suggested modifications to test_barh in test_datetime.py

CozyFrog · CozyFrog · commit fb0802d333af · 2023-10-17T15:45:13.000-04:00
diff --git a/lib/matplotlib/tests/test_datetime.py b/lib/matplotlib/tests/test_datetime.py
@@ -77,17 +77,21 @@ def test_barbs(self):
     @mpl.style.context("default")
     def test_barh(self):
         mpl.rcParams["date.converter"] = 'concise'
-        N = 14
-        base = datetime.datetime(1970, 1, 1)
-        indices = np.arange(N)
-        dates = [base + datetime.timedelta(days=(7 * i)) for i in range(N)]
-        widths = [datetime.timedelta(days=(7*i)) for i in range(N)]
-        fig, (ax1, ax2, ax3, ax4) = plt.subplots(4, 1, layout='constrained')
-        error = np.random.rand(N)
-        ax1.barh(indices, dates, xerr=error)
-        ax2.barh(dates, indices)
-        ax3.barh(dates, dates)
-        ax4.barh(indices, width=widths, left=base)
+        fig, (ax1, ax2) = plt.subplots(2, 1, layout='constrained')
+        birth_date = np.array([datetime.datetime(2020, 4, 10),
+                               datetime.datetime(2020, 5, 30),
+                               datetime.datetime(2020, 10, 12),
+                               datetime.datetime(2020, 11, 15)])
+        year_start = datetime.datetime(2020, 1, 1)
+        year_end = datetime.datetime(2020, 12, 31)
+        age = [21, 53, 20, 24]
+        ax1.set_xlabel('Age')
+        ax1.set_ylabel('Birth Date')
+        ax1.barh(birth_date, width=age, height=datetime.timedelta(days=10))
+        ax2.set_xlim(left=year_start, right=year_end)
+        ax2.set_xlabel('Birth Date')
+        ax2.set_ylabel('Order of Birth Dates')
+        ax2.barh(np.arange(4), birth_date-year_start, left=year_start)
 
     @pytest.mark.xfail(reason="Test for boxplot not written yet")
     @mpl.style.context("default")
