Added smoke test for Axes.imshow to test_datetime.py

ConstableCatnip · ConstableCatnip · commit 3c6d39d48951 · 2023-12-08T21:38:18.000Z
diff --git a/lib/matplotlib/tests/test_datetime.py b/lib/matplotlib/tests/test_datetime.py
@@ -530,11 +530,45 @@ def test_hlines(self):
                          xmin=0.45,
                          xmax=0.65)
 
-    @pytest.mark.xfail(reason="Test for imshow not written yet")
     @mpl.style.context("default")
     def test_imshow(self):
-        fig, ax = plt.subplots()
-        ax.imshow(...)
+        mpl.rcParams["date.converter"] = "concise"
+        a = np.diag(range(5))
+        fig, axes = plt.subplots(nrows=2, ncols=2)
+
+        ax = axes[0, 0]
+        dt_start = datetime.datetime(1980, 4, 15)
+        dt_end = datetime.datetime(2010, 11, 11)
+        extent = (dt_start, dt_end, dt_start, dt_end)
+        ax.imshow(a, extent=extent)
+        for label in ax.get_xticklabels():
+            label.set_rotation(90)
+
+        ax = axes[0, 1]
+        dt_start = datetime.datetime(2010, 4, 15)
+        dt_end = datetime.datetime(2010, 11, 11)
+        extent = (dt_start, dt_end, dt_start, dt_end)
+        ax.imshow(a, extent=extent)
+        for label in ax.get_xticklabels():
+            label.set_rotation(90)
+
+        ax = axes[1, 0]
+        dt_start = datetime.datetime(2010, 11, 1)
+        dt_end = datetime.datetime(2010, 11, 11)
+        extent = (dt_start, dt_end, dt_start, dt_end)
+        ax.imshow(a, extent=extent)
+        for label in ax.get_xticklabels():
+            label.set_rotation(90)
+
+        ax = axes[1, 1]
+        dt_start = datetime.datetime(2010, 11, 10)
+        dt_end = datetime.datetime(2010, 11, 11)
+        extent = (dt_start, dt_end, dt_start, dt_end)
+        ax.imshow(a, extent=extent)
+        for label in ax.get_xticklabels():
+            label.set_rotation(90)
+
+        fig.tight_layout()
 
     @pytest.mark.xfail(reason="Test for loglog not written yet")
     @mpl.style.context("default")
