GH1248/GH1249 PR feedback

Author: loicdiridollou

pandas-stubs/_typing.pyi

@@ -1016,6 +1016,8 @@ IntoColumn: TypeAlias = (
     AnyArrayLike
     | Scalar
     | Callable[[DataFrame], AnyArrayLike | Scalar | list[Scalar] | range]
+    | list[Scalar]
+    | range
     | None
 )
 

tests/test_frame.py

@@ -45,7 +45,9 @@
 )
 import xarray as xr
 
-from pandas._typing import Scalar
+from pandas._typing import (
+    Scalar,
+)
 
 from tests import (
     PD_LTE_23,
@@ -315,11 +317,15 @@ def test_types_assign() -> None:
     df = pd.DataFrame({"a": [1, 2, 3]})
     check(
         assert_type(
-            df.assign(b=lambda df: range(len(df)), c=lambda df: [10, 20, 30]),
+            df.assign(b=lambda df: range(len(df)), c=lambda _: [10, 20, 30]),
             pd.DataFrame,
         ),
         pd.DataFrame,
     )
+    check(
+        assert_type(df.assign(b=range(len(df)), c=[10, 20, 30]), pd.DataFrame),
+        pd.DataFrame,
+    )
 
 
 def test_assign() -> None:

tests/test_indexes.py

@@ -62,8 +62,18 @@ def test_index_astype() -> None:
     )
 
     df = pd.DataFrame({"a": [1, 2, 3]})
-    mi = pd.MultiIndex.from_product([["x", "y"], df.columns])
-    mi = pd.MultiIndex.from_product([["x", "y"], pd.Series([1, 2])])
+    check(
+        assert_type(
+            pd.MultiIndex.from_product([["x", "y"], df.columns]), pd.MultiIndex
+        ),
+        pd.MultiIndex,
+    )
+    check(
+        assert_type(
+            pd.MultiIndex.from_product([["x", "y"], pd.Series([1, 2])]), pd.MultiIndex
+        ),
+        pd.MultiIndex,
+    )
 
 
 def test_multiindex_get_level_values() -> None: