fix index dtype series invert ArrayLike (#168)

Author: Dr-Irv

partial/pandas/_typing.pyi

@@ -36,8 +36,8 @@ if sys.version_info >= (3, 8):
 else:
     from typing_extensions import Literal
 
-AnyArrayLike = TypeVar("AnyArrayLike", ExtensionArray, Index, Series, np.ndarray)
-ArrayLike = TypeVar("ArrayLike", ExtensionArray, np.ndarray)
+ArrayLike = Union[ExtensionArray, np.ndarray]
+AnyArrayLike = Union[Index, Series]
 PythonScalar = Union[str, int, float, bool, complex]
 DatetimeLikeScalar = TypeVar("DatetimeLikeScalar", Period, Timestamp, Timedelta)
 PandasScalar = Union[bytes, datetime.date, datetime.datetime, datetime.timedelta]

partial/pandas/core/indexes/base.pyi

@@ -29,13 +29,14 @@ class Index(IndexOpsMixin[T1], PandasObject, Generic[T1]):
     def __len__(self) -> int: ...
     def __array__(self, dtype=...) -> np.ndarray: ...
     def __array_wrap__(self, result, context=...): ...
-    def dtype(self): ...
+    @property
+    def dtype(self) -> Type[Dtype]: ...
     def ravel(self, order: _str = ...): ...
     def view(self, cls=...): ...
     @overload
     def astype(self, dtype: _str) -> Index: ...
     @overload
-    def astype(self, dtype: Type[T2]) -> Index[T2]: ...
+    def astype(self, dtype: Type[Dtype]) -> Index[Dtype]: ...
     def take(self, indices, axis: int = ..., allow_fill: bool = ..., fill_value=..., **kwargs): ...
     def repeat(self, repeats, axis=...): ...
     def copy(self, name=..., deep: bool = ..., dtype=..., **kwargs): ...

partial/pandas/core/series.pyi

@@ -135,7 +135,7 @@ class Series(IndexOpsMixin, NDFrame, Generic[S1]):
     @name.setter
     def name(self, value: Optional[Hashable]) -> None: ...
     @property
-    def values(self) -> Union[np.ndarray, ExtensionArray]: ...
+    def values(self) -> ArrayLike: ...
     @property
     def array(self) -> ExtensionArray: ...
     def ravel(self, order: _str = ...) -> np.ndarray: ...
@@ -946,6 +946,7 @@ class Series(IndexOpsMixin, NDFrame, Generic[S1]):
     def __sub__(self, other: Union[num, Timedelta, _ListLike, Series[S1]]) -> Series[S1]: ...
     def __truediv__(self, other: Union[num, _ListLike, Series[S1]]) -> Series[S1]: ...
     def __xor__(self, other: Union[_ListLike, Series[S1]]) -> Series: ...
+    def __invert__(self) -> Series[bool]: ...
     # properties
     # @property
     # def array(self) -> _npndarray

tests/pandas/test_indexes.py

@@ -27,3 +27,9 @@ def test_index_isin():
     ind = pd.Index([1, 2, 3, 4, 5])
     isin = ind.isin([2, 4])
     reveal_type(isin, expected_type=np_ndarray_bool)
+
+
+def test_index_astype():
+    indi = pd.Index([1, 2, 3])
+    inds = pd.Index(["a", "b", "c"])
+    indc = indi.astype(inds.dtype)

tests/pandas/test_series.py

@@ -596,3 +596,12 @@ def test_series_index_isin() -> None:
     reveal_type(t2, expected_text="Series[Unknown]")
     reveal_type(t3, expected_text="Series[Unknown]")
     reveal_type(t4, expected_text="Series[Unknown]")
+
+
+def test_series_invert() -> None:
+    s1 = pd.Series([True, False, True])
+    s2 = ~s1
+    reveal_type(s2, expected_text="Series[bool]")
+    s3 = pd.Series([1, 2, 3])
+    reveal_type(s3[s2], expected_type=pd.Series)
+    reveal_type(s3.loc[s2], expected_type=pd.Series)