Typ columns setter as Hashable (#233)

* Typ columns setter as Hashable

* Add tests

* Change type and add tests

* Adjust

Author: phofl

pandas-stubs/core/frame.pyi

@@ -1189,7 +1189,9 @@ class DataFrame(NDFrame, OpsMixin):
     @property
     def columns(self) -> Index: ...
     @columns.setter  # setter needs to be right next to getter; otherwise mypy complains
-    def columns(self, cols: list[_str] | Index[_str]) -> None: ...  # type: ignore[type-arg]
+    def columns(
+        self, cols: AnyArrayLike | list[HashableT] | tuple[HashableT, ...]
+    ) -> None: ...
     @property
     def dtypes(self) -> Series: ...
     @property

tests/test_frame.py

@@ -1408,6 +1408,14 @@ def test_set_columns() -> None:
     # https://github.com/python/mypy/issues/3004
     # pyright doesn't need the ignore
     df.columns = ["c", "d"]  # type: ignore[assignment]
+    df.columns = [1, 2]  # type: ignore[assignment]
+    df.columns = [1, "a"]  # type: ignore[assignment]
+    df.columns = np.array([1, 2])  # type: ignore[assignment]
+    df.columns = pd.Series([1, 2])  # type: ignore[assignment]
+    df.columns = np.array([1, "a"])  # type: ignore[assignment]
+    df.columns = pd.Series([1, "a"])  # type: ignore[assignment]
+    df.columns = (1, 2)  # type: ignore[assignment]
+    df.columns = (1, "a")  # type: ignore[assignment]
 
 
 def test_frame_index_numpy() -> None: