From 688fdc26043641ff5261804fd6e6f3b246075dc2 Mon Sep 17 00:00:00 2001 From: Irv Lustig Date: Fri, 25 Nov 2022 19:33:12 -0500 Subject: [PATCH] Add tuple as possible arguement for axes --- pandas-stubs/_typing.pyi | 2 +- tests/test_frame.py | 8 ++++++++ 2 files changed, 9 insertions(+), 1 deletion(-) diff --git a/pandas-stubs/_typing.pyi b/pandas-stubs/_typing.pyi index 22122ceaf..3ad916f76 100644 --- a/pandas-stubs/_typing.pyi +++ b/pandas-stubs/_typing.pyi @@ -120,7 +120,7 @@ Level: TypeAlias = Union[Hashable, int] Suffixes: TypeAlias = tuple[Optional[str], Optional[str]] Ordered: TypeAlias = Optional[bool] JSONSerializable: TypeAlias = Union[PythonScalar, list, dict] -Axes: TypeAlias = Union[AnyArrayLike, list, dict, range] +Axes: TypeAlias = Union[AnyArrayLike, list, dict, range, tuple] Renamer: TypeAlias = Union[Mapping[Any, Label], Callable[[Any], Label]] T = TypeVar("T") FuncType: TypeAlias = Callable[..., Any] diff --git a/tests/test_frame.py b/tests/test_frame.py index 9e79bef2f..dd2003829 100644 --- a/tests/test_frame.py +++ b/tests/test_frame.py @@ -1953,3 +1953,11 @@ def test_series_groupby_and_value_counts() -> None: ) c: pd.Series = df.groupby("Animal")["Max Speed"].value_counts() check(assert_type(c, pd.Series), pd.Series) + + +def test_axes_as_tuple() -> None: + # GH 384 + index = (3, 5, 7) + columns = ["a", "b", "c"] + df = pd.DataFrame(data=1, index=index, columns=columns) + check(assert_type(df, pd.DataFrame), pd.DataFrame)