add quiverkey datetime test

dkinneer · dkinneer · commit cf2d9c98bb4c · 2023-12-05T16:54:13.000-06:00
diff --git a/lib/matplotlib/tests/test_datetime.py b/lib/matplotlib/tests/test_datetime.py
@@ -520,11 +520,22 @@ def test_quiver(self):
         fig, ax = plt.subplots()
         ax.quiver(...)
 
-    @pytest.mark.xfail(reason="Test for quiverkey not written yet")
     @mpl.style.context("default")
     def test_quiverkey(self):
-        fig, ax = plt.subplots()
-        ax.quiverkey(...)
+        mpl.rcParams["date.converter"] = 'concise'
+        x = np.arange(np.datetime64('2023-12-01'), np.datetime64('2023-12-06'))
+        y = 10 ** np.arange(5)
+        u = np.sin(np.arange(len(x)))
+        v = np.cos(np.arange(len(x)))
+        fig, ax = plt.subplots(figsize=(10, 6))
+        ax.set_title('Quiverplot with Datetime')
+        ax.set_xlabel('Date')
+        ax.set_ylabel('Y-axis')
+        q = ax.quiver(x, y, u, v, scale=20)
+        ax.quiverkey(q, .085, .05, 1, x[0], labelpos='N', coordinates='figure')
+        ax.quiverkey(q, .085, .92, 1, x[1], labelpos='S', coordinates='figure')
+        ax.quiverkey(q, .85, .05, 1, x[2], labelpos='W', coordinates='figure')
+        ax.quiverkey(q, .8, .9, 1, x[3], labelpos='E', coordinates='figure')
 
     @mpl.style.context("default")
     def test_scatter(self):
