Fix matplotlib warnings

pandas-dev · May 26, 2024 · f324108 · f324108
1 parent 16d331c
commit f324108
Showing 1 changed file with 16 additions and 15 deletions.
diff --git a/tests/test_plotting.py b/tests/test_plotting.py
@@ -486,22 +486,23 @@ def test_plot_scatter(close_figures) -> None:
         ),
         Axes,
     )
-    check(
-        assert_type(
-            IRIS_DF.plot.scatter(x="SepalLength", y="SepalWidth", subplots=True),
-            npt.NDArray[np.object_],
-        ),
-        np.ndarray,
-    )
-    check(
-        assert_type(
-            IRIS_DF.plot(
-                x="SepalLength", y="SepalWidth", subplots=True, kind="scatter"
+    with pytest.warns(UserWarning):
+        check(
+            assert_type(
+                IRIS_DF.plot.scatter(x="SepalLength", y="SepalWidth", subplots=True),
+                npt.NDArray[np.object_],
             ),
-            npt.NDArray[np.object_],
-        ),
-        np.ndarray,
-    )
+            np.ndarray,
+        )
+        check(
+            assert_type(
+                IRIS_DF.plot(
+                    x="SepalLength", y="SepalWidth", subplots=True, kind="scatter"
+                ),
+                npt.NDArray[np.object_],
+            ),
+            np.ndarray,
+        )
 
 
 def test_plot_keywords(close_figures):