fixup setitem

Author: MarcoGorelli

pandas-stubs/core/frame.pyi

@@ -866,6 +866,14 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack):
             | None
         ),
     ) -> None: ...
+    # Extra cases not supported by  `_LocIndexerFrame.__setitem__` /
+    # `_iLocIndexerFrame.__setitem__`.
+    @overload
+    def __setitem__(
+        self,
+        idx: IndexOpsMixin | DataFrame,
+        value: Scalar | NAType | NaTType | ArrayLike | Series | list | dict | None,
+    ): ...
     @overload
     def query(
         self,

tests/test_frame.py

@@ -292,6 +292,11 @@ def test_types_setitem() -> None:
     df[5] = [5, 6]
     df[["col1", "col2"]] = [[1, 2], [3, 4]]
     df[s] = [5, 6]
+    df["col1":"col2"] = [5, 6]
+    df.loc[:, s] = [5, 6]
+    df[[0, 1]] = [5, 6]
+    df["col1"] = [5, 6]
+    df[df["col1"] > 1] = [5, 6, 7]
     df[a] = [[1, 2], [3, 4]]
     df[i] = [8, 9]