Merge pull request #27024 from kots14/axes-contourf-unit-test-for-issue-26864

scottshambaugh · web-flow · commit f7e74dad4c74 · 2023-10-19T09:55:35.000-06:00
Add test_contourf in test_datetime.py
diff --git a/lib/matplotlib/tests/test_datetime.py b/lib/matplotlib/tests/test_datetime.py
@@ -123,11 +123,29 @@ def test_contour(self):
         fig, ax = plt.subplots()
         ax.contour(...)
 
-    @pytest.mark.xfail(reason="Test for contourf not written yet")
     @mpl.style.context("default")
     def test_contourf(self):
-        fig, ax = plt.subplots()
-        ax.contourf(...)
+        mpl.rcParams["date.converter"] = "concise"
+        range_threshold = 10
+        fig, (ax1, ax2, ax3) = plt.subplots(3, 1, layout="constrained")
+
+        x_dates = np.array(
+            [datetime.datetime(2023, 10, delta) for delta in range(1, range_threshold)]
+        )
+        y_dates = np.array(
+            [datetime.datetime(2023, 10, delta) for delta in range(1, range_threshold)]
+        )
+        x_ranges = np.array(range(1, range_threshold))
+        y_ranges = np.array(range(1, range_threshold))
+
+        X_dates, Y_dates = np.meshgrid(x_dates, y_dates)
+        X_ranges, Y_ranges = np.meshgrid(x_ranges, y_ranges)
+
+        Z_ranges = np.cos(X_ranges / 4) + np.sin(Y_ranges / 4)
+
+        ax1.contourf(X_dates, Y_dates, Z_ranges)
+        ax2.contourf(X_dates, Y_ranges, Z_ranges)
+        ax3.contourf(X_ranges, Y_dates, Z_ranges)
 
     @pytest.mark.xfail(reason="Test for errorbar not written yet")
     @mpl.style.context("default")
