Add test cases for {DataFrame,Series}#get(..., default=None) would return None

Author: skatsuta

tests/test_frame.py

@@ -3120,6 +3120,7 @@ def test_get() -> None:
         pd.Series,
         np.int64,
     )
+    check(assert_type(df.get("z", default=None), Union[pd.Series, None]), type(None))
     check(
         assert_type(df.get("a", default=1), Union[pd.Series, int]), pd.Series, np.int64
     )
@@ -3133,6 +3134,9 @@ def test_get() -> None:
         assert_type(df.get(["a", "b"], default=None), Union[pd.DataFrame, None]),
         pd.DataFrame,
     )
+    check(
+        assert_type(df.get(["z"], default=None), Union[pd.DataFrame, None]), type(None)
+    )
     check(
         assert_type(df.get(["a", "b"], default=1), Union[pd.DataFrame, int]),
         pd.DataFrame,

tests/test_series.py

@@ -2882,6 +2882,7 @@ def test_get() -> None:
     check(assert_type(s_int.get(1), Union[int, None]), np.int64)
     check(assert_type(s_int.get(99), Union[int, None]), type(None))
     check(assert_type(s_int.get(1, default=None), Union[int, None]), np.int64)
+    check(assert_type(s_int.get(99, default=None), Union[int, None]), type(None))
     check(assert_type(s_int.get(1, default=2), int), np.int64)
     check(assert_type(s_int.get(99, default="a"), Union[int, str]), str)
 
@@ -2890,6 +2891,7 @@ def test_get() -> None:
     check(assert_type(s_str.get("a"), Union[str, None]), str)
     check(assert_type(s_str.get("z"), Union[str, None]), type(None))
     check(assert_type(s_str.get("a", default=None), Union[str, None]), str)
+    check(assert_type(s_str.get("z", default=None), Union[str, None]), type(None))
     check(assert_type(s_str.get("a", default="b"), str), str)
     check(assert_type(s_str.get("z", default=True), Union[str, bool]), bool)