GH1396 Test out

Author: loicdiridollou

pandas-stubs/core/reshape/concat.pyi

@@ -6,12 +6,9 @@ from collections.abc import (
 from typing import (
     Literal,
     Never,
-    TypeVar,
     overload,
 )
 
-from pandas.core.frame import DataFrame
-from pandas.core.generic import NDFrame
 from pandas.core.series import Series
 
 from pandas._typing import (
@@ -22,10 +19,9 @@ from pandas._typing import (
     HashableT2,
     HashableT3,
     HashableT4,
+    NDFrameT,
 )
 
-DataFrameT0 = TypeVar("DataFrameT0", bound=DataFrame)
-
 @overload
 def concat(
     objs: Iterable[None] | Mapping[HashableT1, None],
@@ -40,7 +36,7 @@ def concat(
     sort: bool = False,
 ) -> Never: ...
 @overload
-def concat(  # type: ignore[overload-overlap] # pyright: ignore[reportOverlappingOverload]
+def concat(
     objs: Iterable[Series[S2] | None] | Mapping[HashableT1, Series[S2] | None],
     *,
     axis: AxisIndex = 0,
@@ -67,20 +63,7 @@ def concat(  # type: ignore[overload-overlap]
 ) -> Series: ...
 @overload
 def concat(
-    objs: Iterable[DataFrameT0 | None],
-    *,
-    axis: Axis = 0,
-    join: Literal["inner", "outer"] = "outer",
-    ignore_index: bool = False,
-    keys: Iterable[HashableT2] | None = None,
-    levels: Sequence[list[HashableT3] | tuple[HashableT3, ...]] | None = None,
-    names: list[HashableT4] | None = None,
-    verify_integrity: bool = False,
-    sort: bool = False,
-) -> DataFrameT0: ...
-@overload
-def concat(
-    objs: Iterable[NDFrame | None] | Mapping[HashableT1, NDFrame | None],
+    objs: Iterable[NDFrameT | None] | Mapping[HashableT1, NDFrameT | None],
     *,
     axis: Axis = 0,
     join: Literal["inner", "outer"] = "outer",
@@ -90,4 +73,18 @@ def concat(
     names: list[HashableT4] | None = None,
     verify_integrity: bool = False,
     sort: bool = False,
-) -> DataFrame: ...
+) -> NDFrameT: ...
+
+# @overload
+# def concat(
+#     objs: Iterable[NDFrame | None] | Mapping[HashableT1, NDFrame | None],
+#     *,
+#     axis: Axis = 0,
+#     join: Literal["inner", "outer"] = "outer",
+#     ignore_index: bool = False,
+#     keys: Iterable[HashableT2] | None = None,
+#     levels: Sequence[list[HashableT3] | tuple[HashableT3, ...]] | None = None,
+#     names: list[HashableT4] | None = None,
+#     verify_integrity: bool = False,
+#     sort: bool = False,
+# ) -> DataFrame: ...

tests/test_pandas.py

@@ -9,6 +9,7 @@
     Any,
     Never,
     assert_type,
+    reveal_type,
 )
 
 import numpy as np