maint: split Sequence[int] out from AnyArrayLikeInt (#1425)

* maint: split `Sequence[int]` out from `AnyArrayLikeInt`

* add ndarray test

* use npt.NDArray instead of 1d numpy array

Author: MarcoGorelli

pandas-stubs/_typing.pyi

@@ -86,6 +86,9 @@ HashableT5 = TypeVar("HashableT5", bound=Hashable)
 
 ArrayLike: TypeAlias = ExtensionArray | np.ndarray
 AnyArrayLike: TypeAlias = ArrayLike | Index | Series
+AnyArrayLikeInt: TypeAlias = (
+    IntegerArray | Index[int] | Series[int] | npt.NDArray[np.integer]
+)
 
 # list-like
 
@@ -864,10 +867,6 @@ np_ndarray: TypeAlias = np.ndarray[ShapeT, np.dtype[GenericT]]
 np_1darray: TypeAlias = np.ndarray[tuple[int], np.dtype[GenericT]]
 np_2darray: TypeAlias = np.ndarray[tuple[int, int], np.dtype[GenericT]]
 
-AnyArrayLikeInt: TypeAlias = (
-    IntegerArray | Index[int] | Series[int] | np_1darray[np.integer] | Sequence[int]
-)
-
 class SupportsDType(Protocol[GenericT_co]):
     @property
     def dtype(self) -> np.dtype[GenericT_co]: ...

pandas-stubs/core/arrays/base.pyi

@@ -1,4 +1,7 @@
-from collections.abc import Iterator
+from collections.abc import (
+    Iterator,
+    Sequence,
+)
 from typing import (
     Any,
     Literal,
@@ -58,7 +61,9 @@ class ExtensionArray:
     def unique(self) -> Self: ...
     def searchsorted(self, value, side: str = ..., sorter=...): ...
     def factorize(self, use_na_sentinel: bool = True) -> tuple[np_1darray, Self]: ...
-    def repeat(self, repeats: int | AnyArrayLikeInt, axis: None = None) -> Self: ...
+    def repeat(
+        self, repeats: int | AnyArrayLikeInt | Sequence[int], axis: None = None
+    ) -> Self: ...
     def take(
         self,
         indexer: TakeIndexer,

tests/arrays/test_extension_array.py

@@ -1,5 +1,5 @@
 # Test common ExtensionArray methods
-
+import numpy as np
 import pandas as pd
 from pandas.core.arrays.integer import IntegerArray
 from pandas.core.construction import array
@@ -14,6 +14,7 @@ def test_ea_common() -> None:
 
     check(assert_type(arr.repeat(1), IntegerArray), IntegerArray)
     check(assert_type(arr.repeat(arr), IntegerArray), IntegerArray)
+    check(assert_type(arr.repeat(np.array([1, 2, 3])), IntegerArray), IntegerArray)
     check(
         assert_type(arr.repeat(repeats=pd.Series([1, 2, 3])), IntegerArray),
         IntegerArray,