diff --git a/pandas-stubs/core/frame.pyi b/pandas-stubs/core/frame.pyi index a9e2d58da..664803b81 100644 --- a/pandas-stubs/core/frame.pyi +++ b/pandas-stubs/core/frame.pyi @@ -542,7 +542,7 @@ class DataFrame(NDFrame, OpsMixin): self, loc: int, column, - value: int | ListLike, + value: Scalar | ListLikeU | None, allow_duplicates: _bool = ..., ) -> None: ... def assign(self, **kwargs) -> DataFrame: ... diff --git a/tests/test_frame.py b/tests/test_frame.py index 5af0390bd..eeb6dd924 100644 --- a/tests/test_frame.py +++ b/tests/test_frame.py @@ -2410,3 +2410,12 @@ def test_in_columns() -> None: check(assert_type(df.loc[:, cols], pd.DataFrame), pd.DataFrame) check(assert_type(df[cols], pd.DataFrame), pd.DataFrame) check(assert_type(df.groupby(by=cols).sum(), pd.DataFrame), pd.DataFrame) + + +def test_insert_newvalues() -> None: + df = pd.DataFrame({"a": [1, 2]}) + ab = pd.DataFrame({"col1": [1, 2], "col2": [3, 4]}) + ef = pd.DataFrame({"z": [4, 5, 6]}) + assert assert_type(df.insert(loc=0, column="b", value=None), None) is None + assert assert_type(ab.insert(loc=0, column="newcol", value=[99, 99]), None) is None + assert assert_type(ef.insert(loc=0, column="g", value=4), None) is None