move private imports from testing into tests/__init__.py (#1205)

Author: Dr-Irv

pandas-stubs/_typing.pyi

@@ -31,6 +31,7 @@ from pandas.core.generic import NDFrame
 from pandas.core.groupby.grouper import Grouper
 from pandas.core.indexes.base import Index
 from pandas.core.series import Series
+from pandas.core.tools.datetimes import FulldatetimeDict
 from typing_extensions import (
     ParamSpec,
     TypeAlias,
@@ -73,22 +74,6 @@ DatetimeDictArg: TypeAlias = (
 )
 DictConvertible: TypeAlias = FulldatetimeDict | DataFrame
 
-class YearMonthDayDict(TypedDict, total=True):
-    year: DatetimeDictArg
-    month: DatetimeDictArg
-    day: DatetimeDictArg
-
-class FulldatetimeDict(YearMonthDayDict, total=False):
-    hour: DatetimeDictArg
-    hours: DatetimeDictArg
-    minute: DatetimeDictArg
-    minutes: DatetimeDictArg
-    second: DatetimeDictArg
-    seconds: DatetimeDictArg
-    ms: DatetimeDictArg
-    us: DatetimeDictArg
-    ns: DatetimeDictArg
-
 CorrelationMethod: TypeAlias = (
     Literal["pearson", "kendall", "spearman"]
     | Callable[[np.ndarray, np.ndarray], float]

pandas-stubs/core/tools/datetimes.pyi

@@ -5,6 +5,7 @@ from datetime import (
 )
 from typing import (
     Literal,
+    TypedDict,
     overload,
 )
 
@@ -37,6 +38,24 @@ DatetimeScalar: TypeAlias = Scalar | datetime | np.datetime64 | date
 
 DatetimeScalarOrArrayConvertible: TypeAlias = DatetimeScalar | ArrayConvertible
 
+DatetimeDictArg: TypeAlias = list[Scalar] | tuple[Scalar, ...] | AnyArrayLike
+
+class YearMonthDayDict(TypedDict, total=True):
+    year: DatetimeDictArg
+    month: DatetimeDictArg
+    day: DatetimeDictArg
+
+class FulldatetimeDict(YearMonthDayDict, total=False):
+    hour: DatetimeDictArg
+    hours: DatetimeDictArg
+    minute: DatetimeDictArg
+    minutes: DatetimeDictArg
+    second: DatetimeDictArg
+    seconds: DatetimeDictArg
+    ms: DatetimeDictArg
+    us: DatetimeDictArg
+    ns: DatetimeDictArg
+
 @overload
 def to_datetime(
     arg: DatetimeScalar,

tests/__init__.py

@@ -14,12 +14,39 @@
 )
 
 import numpy as np
+import numpy.typing as npt
 import pandas as pd
+
+# Next set of imports is to keep the private imports needed for testing
+# in one place
+from pandas._testing import ensure_clean as ensure_clean
 from pandas.core.groupby.groupby import BaseGroupBy
 from pandas.util.version import Version
 import pytest
 
-from pandas._typing import T
+if TYPE_CHECKING:
+    from pandas._typing import (
+        BooleanDtypeArg as BooleanDtypeArg,
+        BytesDtypeArg as BytesDtypeArg,
+        CategoryDtypeArg as CategoryDtypeArg,
+        ComplexDtypeArg as ComplexDtypeArg,
+        Dtype as Dtype,
+        FloatDtypeArg as FloatDtypeArg,
+        IntDtypeArg as IntDtypeArg,
+        ObjectDtypeArg as ObjectDtypeArg,
+        StrDtypeArg as StrDtypeArg,
+        T as T,
+        TimedeltaDtypeArg as TimedeltaDtypeArg,
+        TimestampDtypeArg as TimestampDtypeArg,
+        UIntDtypeArg as UIntDtypeArg,
+        VoidDtypeArg as VoidDtypeArg,
+        np_ndarray_bool as np_ndarray_bool,
+        np_ndarray_int as np_ndarray_int,
+    )
+else:
+    # Separately define here so pytest works
+    np_ndarray_bool = npt.NDArray[np.bool_]
+    np_ndarray_int = npt.NDArray[np.signedinteger]
 
 TYPE_CHECKING_INVALID_USAGE: Final = TYPE_CHECKING
 WINDOWS = os.name == "nt" or "cygwin" in platform.system().lower()

tests/extension/decimal/array.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+from builtins import type as type_t
 import decimal
 import numbers
 import sys
@@ -24,7 +25,6 @@
 
 from pandas._typing import (
     TakeIndexer,
-    type_t,
 )
 
 from pandas.core.dtypes.base import ExtensionDtype

tests/test_api_typing.py

@@ -2,7 +2,7 @@
 
 import numpy as np
 import pandas as pd
-from pandas._testing import ensure_clean
+from pandas import read_json
 from pandas.api.typing import (
     DataFrameGroupBy,
     DatetimeIndexResamplerGroupby,
@@ -29,9 +29,10 @@
     assert_type,
 )
 
-from tests import check
-
-from pandas.io.json._json import read_json
+from tests import (
+    check,
+    ensure_clean,
+)
 
 ResamplerGroupBy: TypeAlias = (
     DatetimeIndexResamplerGroupby

tests/test_dtypes.py

@@ -14,15 +14,17 @@
 import numpy as np
 import pandas as pd
 from pandas.api.types import is_any_real_numeric_dtype
+from pandas.api.typing import (
+    NaTType,
+    NAType,
+)
 from pandas.core.arrays import (
     BooleanArray,
     IntegerArray,
 )
 import pyarrow as pa
 from typing_extensions import assert_type
 
-from pandas._libs import NaTType
-from pandas._libs.missing import NAType
 from pandas._typing import Scalar
 
 from tests import (

tests/test_frame.py

@@ -31,7 +31,8 @@
 import numpy as np
 import numpy.typing as npt
 import pandas as pd
-from pandas._testing import ensure_clean
+from pandas import Timestamp
+from pandas.api.typing import NAType
 from pandas.core.resample import (
     DatetimeIndexResampler,
     Resampler,
@@ -44,14 +45,13 @@
 )
 import xarray as xr
 
-from pandas._libs.missing import NAType
-from pandas._libs.tslibs.timestamps import Timestamp
 from pandas._typing import Scalar
 
 from tests import (
     PD_LTE_22,
     TYPE_CHECKING_INVALID_USAGE,
     check,
+    ensure_clean,
     pytest_warns_bounded,
 )
 

tests/test_indexes.py

@@ -1,7 +1,10 @@
 from __future__ import annotations
 
 import datetime as dt
-from typing import Union
+from typing import (
+    TYPE_CHECKING,
+    Union,
+)
 
 import numpy as np
 from numpy import typing as npt
@@ -13,7 +16,8 @@
     assert_type,
 )
 
-from pandas._typing import Dtype  # noqa: F401
+if TYPE_CHECKING:
+    from tests import Dtype  # noqa: F401
 
 from tests import (
     PD_LTE_22,

tests/test_io.py

@@ -40,7 +40,7 @@
     read_table,
     read_xml,
 )
-from pandas._testing import ensure_clean
+from pandas.api.typing import JsonReader
 import pytest
 import sqlalchemy
 import sqlalchemy.orm
@@ -51,11 +51,11 @@
     TYPE_CHECKING_INVALID_USAGE,
     WINDOWS,
     check,
+    ensure_clean,
 )
 from tests import NUMPY20  # See https://github.com/PyTables/PyTables/issues/1172
 
 from pandas.io.api import to_pickle
-from pandas.io.json._json import JsonReader
 from pandas.io.parsers import TextFileReader
 from pandas.io.pytables import (
     TableIterator,

tests/test_pandas.py

@@ -13,6 +13,10 @@
 import pandas as pd
 from pandas import Grouper
 from pandas.api.extensions import ExtensionArray
+from pandas.api.typing import (
+    NaTType,
+    NAType,
+)
 import pandas.util as pdutil
 
 # TODO: github.com/pandas-dev/pandas/issues/55023
@@ -23,8 +27,6 @@
     assert_type,
 )
 
-from pandas._libs.missing import NAType
-from pandas._libs.tslibs import NaTType
 from pandas._typing import Scalar
 
 from tests import (