use assert_type instead

Author: hyperc54

tests/test_frame.py

@@ -373,12 +373,21 @@ def test_arguments_drop() -> None:
 
 def test_types_dropna() -> None:
     df = pd.DataFrame(data={"col1": [np.nan, np.nan], "col2": [3, np.nan]})
-    res: pd.DataFrame = df.dropna()
-    res2: pd.DataFrame = df.dropna(ignore_index=True)
-    res3: pd.DataFrame = df.dropna(axis=1, thresh=1)
-    res4: None = df.dropna(axis=0, how="all", subset=["col1"], inplace=True)
-    res5: None = df.dropna(
-        axis=0, how="all", subset=["col1"], inplace=True, ignore_index=False
+    check(assert_type(df.dropna(), pd.DataFrame), pd.DataFrame)
+    check(assert_type(df.dropna(ignore_index=True), pd.DataFrame), pd.DataFrame)
+    check(assert_type(df.dropna(axis=1, thresh=1), pd.DataFrame), pd.DataFrame)
+    assert (
+        assert_type(df.dropna(axis=0, how="all", subset=["col1"], inplace=True), None)
+        is None
+    )
+    assert (
+        assert_type(
+            df.dropna(
+                axis=0, how="all", subset=["col1"], inplace=True, ignore_index=False
+            ),
+            None,
+        )
+        is None
     )