use pd.array directly

Author: cmp0xff

pandas-stubs/core/construction.pyi

@@ -1,10 +1,21 @@
 from collections.abc import Sequence
+from typing import overload
 
 import numpy as np
 from pandas.core.arrays.base import ExtensionArray
+from pandas.core.arrays.integer import IntegerArray
+
+from pandas._libs.missing import NAType
 
 from pandas.core.dtypes.dtypes import ExtensionDtype
 
+@overload
+def array(
+    data: Sequence[int] | Sequence[int | NAType],
+    dtype: str | np.dtype | ExtensionDtype | None = None,
+    copy: bool = True,
+) -> IntegerArray: ...
+@overload
 def array(
     data: Sequence[object],
     dtype: str | np.dtype | ExtensionDtype | None = None,

tests/arrays/test_arrays.py

@@ -1,39 +1,12 @@
-from typing import cast
-
-from pandas.core.arrays.integer import (
-    Int64Dtype,
-    IntegerArray,
-)
-from pandas.core.series import Series
+from pandas.core.arrays.integer import IntegerArray
+from pandas.core.construction import array
 from typing_extensions import assert_type
 
 from pandas._libs.missing import NA
 
 from tests import check
 
 
-def test_cumul_int64dtype():
-    # cast will be removed if pandas-dev/pandas-stubs#1395 is resolved
-    arr = cast("Series[Int64Dtype]", Series([1, NA, 2], dtype="Int64")).array
-
-    check(assert_type(arr._accumulate("cummin"), IntegerArray), IntegerArray)
-    check(assert_type(arr._accumulate("cummax"), IntegerArray), IntegerArray)
-    check(assert_type(arr._accumulate("cumsum"), IntegerArray), IntegerArray)
-    check(assert_type(arr._accumulate("cumprod"), IntegerArray), IntegerArray)
-
-    check(
-        assert_type(arr._accumulate("cummin", skipna=False), IntegerArray),
-        IntegerArray,
-    )
-    check(
-        assert_type(arr._accumulate("cummax", skipna=False), IntegerArray),
-        IntegerArray,
-    )
-    check(
-        assert_type(arr._accumulate("cumsum", skipna=False), IntegerArray),
-        IntegerArray,
-    )
-    check(
-        assert_type(arr._accumulate("cumprod", skipna=False), IntegerArray),
-        IntegerArray,
-    )
+def test_construction() -> None:
+    check(assert_type(array([1]), IntegerArray), IntegerArray)
+    check(assert_type(array([1, NA]), IntegerArray), IntegerArray)

tests/arrays/test_cumul.py

@@ -0,0 +1,33 @@
+from pandas.core.arrays.integer import IntegerArray
+from pandas.core.construction import array
+from typing_extensions import assert_type
+
+from pandas._libs.missing import NA
+
+from tests import check
+
+
+def test_cumul_int64dtype() -> None:
+    arr = array([1, NA, 2])
+
+    check(assert_type(arr._accumulate("cummin"), IntegerArray), IntegerArray)
+    check(assert_type(arr._accumulate("cummax"), IntegerArray), IntegerArray)
+    check(assert_type(arr._accumulate("cumsum"), IntegerArray), IntegerArray)
+    check(assert_type(arr._accumulate("cumprod"), IntegerArray), IntegerArray)
+
+    check(
+        assert_type(arr._accumulate("cummin", skipna=False), IntegerArray),
+        IntegerArray,
+    )
+    check(
+        assert_type(arr._accumulate("cummax", skipna=False), IntegerArray),
+        IntegerArray,
+    )
+    check(
+        assert_type(arr._accumulate("cumsum", skipna=False), IntegerArray),
+        IntegerArray,
+    )
+    check(
+        assert_type(arr._accumulate("cumprod", skipna=False), IntegerArray),
+        IntegerArray,
+    )