GH1520 Clean up klass for npt.NDArray[np.object_] (#1552)

Author: loicdiridollou

tests/test_plotting.py

@@ -212,10 +212,10 @@ def test_bootstrap_plot(close_figures: None) -> None:
 
 def test_hist(close_figures: None) -> None:
     df = pd.DataFrame({"Col1": [1, 2], "Col4": [2, 1]})
-    check(assert_type(df.hist(), npt.NDArray[np.object_]), npt.NDArray[np.object_])
+    check(assert_type(df.hist(), npt.NDArray[np.object_]), np.ndarray)
     check(
         assert_type(df.hist(by="Col4"), npt.NDArray[np.object_]),
-        npt.NDArray[np.object_],
+        np.ndarray,
     )
 
 
@@ -380,7 +380,7 @@ def test_scatter_matrix(close_figures: None) -> None:
             pd.plotting.scatter_matrix(df, alpha=0.2),
             npt.NDArray[np.object_],
         ),
-        npt.NDArray[np.object_],
+        np.ndarray,
     )
 
 
@@ -399,14 +399,14 @@ def test_plot_line() -> None:
             IRIS_DF.plot.line(subplots=True),
             npt.NDArray[np.object_],
         ),
-        npt.NDArray[np.object_],
+        np.ndarray,
     )
     check(
         assert_type(
             IRIS_DF.plot(kind="line", subplots=True),
             npt.NDArray[np.object_],
         ),
-        npt.NDArray[np.object_],
+        np.ndarray,
     )
 
 
@@ -418,14 +418,14 @@ def test_plot_area(close_figures: None) -> None:
             IRIS_DF.plot.area(subplots=True),
             npt.NDArray[np.object_],
         ),
-        npt.NDArray[np.object_],
+        np.ndarray,
     )
     check(
         assert_type(
             IRIS_DF.plot(kind="area", subplots=True),
             npt.NDArray[np.object_],
         ),
-        npt.NDArray[np.object_],
+        np.ndarray,
     )
 
 
@@ -437,14 +437,14 @@ def test_plot_bar(close_figures: None) -> None:
             IRIS_DF.plot.bar(subplots=True),
             npt.NDArray[np.object_],
         ),
-        npt.NDArray[np.object_],
+        np.ndarray,
     )
     check(
         assert_type(
             IRIS_DF.plot(kind="bar", subplots=True),
             npt.NDArray[np.object_],
         ),
-        npt.NDArray[np.object_],
+        np.ndarray,
     )
 
 
@@ -456,14 +456,14 @@ def test_plot_barh(close_figures: None) -> None:
             IRIS_DF.plot.barh(subplots=True),
             npt.NDArray[np.object_],
         ),
-        npt.NDArray[np.object_],
+        np.ndarray,
     )
     check(
         assert_type(
             IRIS_DF.plot(kind="barh", subplots=True),
             npt.NDArray[np.object_],
         ),
-        npt.NDArray[np.object_],
+        np.ndarray,
     )
 
 
@@ -494,14 +494,14 @@ def test_plot_density(close_figures: None) -> None:
             IRIS_DF.plot.density(subplots=True),
             npt.NDArray[np.object_],
         ),
-        npt.NDArray[np.object_],
+        np.ndarray,
     )
     check(
         assert_type(
             IRIS_DF.plot(kind="density", subplots=True),
             npt.NDArray[np.object_],
         ),
-        npt.NDArray[np.object_],
+        np.ndarray,
     )
 
 
@@ -519,14 +519,14 @@ def test_plot_hexbin(close_figures: None) -> None:
             IRIS_DF.plot.hexbin(x="SepalLength", y="SepalWidth", subplots=True),
             npt.NDArray[np.object_],
         ),
-        npt.NDArray[np.object_],
+        np.ndarray,
     )
     check(
         assert_type(
             IRIS_DF.plot(kind="hexbin", x="SepalLength", y="SepalWidth", subplots=True),
             npt.NDArray[np.object_],
         ),
-        npt.NDArray[np.object_],
+        np.ndarray,
     )
 
 
@@ -538,14 +538,14 @@ def test_plot_hist(close_figures: None) -> None:
             IRIS_DF.plot.hist(subplots=True),
             npt.NDArray[np.object_],
         ),
-        npt.NDArray[np.object_],
+        np.ndarray,
     )
     check(
         assert_type(
             IRIS_DF.plot(subplots=True, kind="hist"),
             npt.NDArray[np.object_],
         ),
-        npt.NDArray[np.object_],
+        np.ndarray,
     )
 
 
@@ -557,14 +557,14 @@ def test_plot_kde(close_figures: None) -> None:
             IRIS_DF.plot.kde(subplots=True),
             npt.NDArray[np.object_],
         ),
-        npt.NDArray[np.object_],
+        np.ndarray,
     )
     check(
         assert_type(
             IRIS_DF.plot(subplots=True, kind="kde"),
             npt.NDArray[np.object_],
         ),
-        npt.NDArray[np.object_],
+        np.ndarray,
     )
 
 
@@ -576,15 +576,15 @@ def test_plot_pie(close_figures: None) -> None:
             IRIS_DF.plot.pie(y="SepalLength", subplots=True),
             npt.NDArray[np.object_],
         ),
-        npt.NDArray[np.object_],
+        np.ndarray,
     )
 
     check(
         assert_type(
             IRIS_DF.plot(kind="pie", y="SepalLength", subplots=True),
             npt.NDArray[np.object_],
         ),
-        npt.NDArray[np.object_],
+        np.ndarray,
     )
 
 
@@ -604,7 +604,7 @@ def test_plot_scatter(close_figures: None) -> None:
             IRIS_DF.plot.scatter(x="SepalLength", y="SepalWidth", subplots=True),
             npt.NDArray[np.object_],
         ),
-        npt.NDArray[np.object_],
+        np.ndarray,
     )
     check(
         assert_type(
@@ -613,7 +613,7 @@ def test_plot_scatter(close_figures: None) -> None:
             ),
             npt.NDArray[np.object_],
         ),
-        npt.NDArray[np.object_],
+        np.ndarray,
     )
 
 
@@ -697,7 +697,7 @@ def test_plot_subplot_changes_150() -> None:
         assert_type(
             df.plot(subplots=[("a", "b"), ("c", "d")]), npt.NDArray[np.object_]
         ),
-        npt.NDArray[np.object_],
+        np.ndarray,
     )