Skip to content

Commit 5d1c228

Browse files
cmp0xffcmp0xff
committed
series
1 parent ea5d6bf commit 5d1c228

File tree

5 files changed

+26
-56
lines changed

5 files changed

+26
-56
lines changed

‎pandas-stubs/core/indexes/base.pyi‎

Lines changed: 5 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,9 @@ from pandas._typing import (
115115
from pandas.core.dtypes.dtypes import PeriodDtype
116116

117117
NumpyRealScalar: TypeAlias = np.bool | np.integer | np.floating
118-
IndexReal: TypeAlias = Index[bool] | Index[int] | Index[float] # ty: ignore[unresolved-reference]
118+
IndexReal: TypeAlias = (
119+
Index[bool] | Index[int] | Index[float] # ty: ignore[unresolved-reference]
120+
)
119121
ScalarArrayIndexReal: TypeAlias = (
120122
float
121123
| Sequence[float | NumpyRealScalar]
@@ -1086,19 +1088,7 @@ class Index(IndexOpsMixin[S1], ElementOpsMixin[S1]):
10861088
@overload
10871089
def __rtruediv__(self, other: Path) -> Index: ...
10881090
@overload
1089-
def __floordiv__(
1090-
self: Index[Never],
1091-
other: (
1092-
float
1093-
| Sequence[float]
1094-
| np_ndarray_bool
1095-
| np_ndarray_anyint
1096-
| np_ndarray_float
1097-
| Index[bool]
1098-
| Index[int]
1099-
| Index[float]
1100-
),
1101-
) -> Index: ...
1091+
def __floordiv__(self: Index[Never], other: ScalarArrayIndexReal) -> Index: ...
11021092
@overload
11031093
def __floordiv__(
11041094
self: Index[bool] | Index[int] | Index[float], other: Index[Never]
@@ -1139,19 +1129,7 @@ class Index(IndexOpsMixin[S1], ElementOpsMixin[S1]):
11391129
other: float | Sequence[float] | np_ndarray_float | Index[float],
11401130
) -> Index[float]: ...
11411131
@overload
1142-
def __rfloordiv__(
1143-
self: Index[Never],
1144-
other: (
1145-
float
1146-
| Sequence[float]
1147-
| np_ndarray_bool
1148-
| np_ndarray_anyint
1149-
| np_ndarray_float
1150-
| Index[bool]
1151-
| Index[int]
1152-
| Index[float]
1153-
),
1154-
) -> Index: ...
1132+
def __rfloordiv__(self: Index[Never], other: ScalarArrayIndexReal) -> Index: ...
11551133
@overload
11561134
def __rfloordiv__( # type: ignore[overload-overlap]
11571135
self: Index[bool] | Index[int] | Index[float], other: Index[Never]

‎pandas-stubs/core/series.pyi‎

Lines changed: 18 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -309,11 +309,17 @@ _DataLikeS1: TypeAlias = (
309309
ArrayLike | dict[_str, np.ndarray] | Sequence[S1] | IndexOpsMixin[S1]
310310
)
311311

312-
SeriesReal: TypeAlias = Series[bool] | Series[int] | Series[float] # ty: ignore[unresolved-reference]
312+
SeriesReal: TypeAlias = (
313+
Series[bool] | Series[int] | Series[float] # ty: ignore[unresolved-reference]
314+
)
313315
ScalarArrayIndexSeriesReal: TypeAlias = ScalarArrayIndexReal | SeriesReal
314-
SeriesComplex: TypeAlias = SeriesReal | Series[complex] # ty: ignore[unresolved-reference]
316+
SeriesComplex: TypeAlias = (
317+
SeriesReal | Series[complex] # ty: ignore[unresolved-reference]
318+
)
315319
ScalarArrayIndexSeriesComplex: TypeAlias = ScalarArrayIndexComplex | SeriesComplex
316-
ArrayIndexSeriesTimedeltaNoSeq: TypeAlias = ArrayIndexTimedeltaNoSeq | Series[Timedelta] # ty: ignore[unresolved-reference]
320+
ArrayIndexSeriesTimedeltaNoSeq: TypeAlias = (
321+
ArrayIndexTimedeltaNoSeq | Series[Timedelta] # ty: ignore[unresolved-reference]
322+
)
317323
ScalarArrayIndexSeriesTimedelta: TypeAlias = (
318324
ScalarArrayIndexTimedelta | Series[Timedelta] # ty: ignore[unresolved-reference]
319325
)
@@ -2404,6 +2410,11 @@ class Series(IndexOpsMixin[S1], ElementOpsMixin[S1], NDFrame):
24042410
other: timedelta | np.timedelta64 | ArrayIndexSeriesTimedeltaNoSeq,
24052411
) -> Series[Timedelta]: ...
24062412
@overload
2413+
def __rfloordiv__(
2414+
self: Series[int] | Series[float],
2415+
other: Sequence[timedelta | np.timedelta64],
2416+
) -> Series: ...
2417+
@overload
24072418
def __rfloordiv__(
24082419
self: Series[Timedelta], other: ArrayIndexSeriesTimedeltaNoSeq
24092420
) -> Series[int]: ...
@@ -2476,21 +2487,15 @@ class Series(IndexOpsMixin[S1], ElementOpsMixin[S1], NDFrame):
24762487
@overload
24772488
def rfloordiv(
24782489
self: Series[int] | Series[float],
2479-
other: (
2480-
timedelta
2481-
| Sequence[timedelta]
2482-
| np_ndarray_td
2483-
| TimedeltaIndex
2484-
| Series[Timedelta]
2485-
),
2490+
other: ScalarArrayIndexSeriesTimedelta,
24862491
level: Level | None = ...,
24872492
fill_value: float | None = None,
24882493
axis: AxisIndex = ...,
24892494
) -> Series[Timedelta]: ...
24902495
@overload
24912496
def rfloordiv(
24922497
self: Series[Timedelta],
2493-
other: ArrayIndexSeriesTimedeltaNoSeq,
2498+
other: timedelta | np.timedelta64 | ArrayIndexSeriesTimedeltaNoSeq,
24942499
level: Level | None = ...,
24952500
fill_value: float | None = None,
24962501
axis: AxisIndex = ...,
@@ -3967,14 +3972,7 @@ class Series(IndexOpsMixin[S1], ElementOpsMixin[S1], NDFrame):
39673972
) -> Series[float]: ...
39683973
@overload
39693974
def __rtruediv__(
3970-
self: Series[int] | Series[float],
3971-
other: (
3972-
timedelta
3973-
| Sequence[timedelta]
3974-
| np_ndarray_td
3975-
| TimedeltaIndex
3976-
| Series[Timedelta]
3977-
),
3975+
self: Series[int] | Series[float], other: ScalarArrayIndexSeriesTimedelta
39783976
) -> Series[Timedelta]: ...
39793977
@overload
39803978
def __rtruediv__(self: Series[_str], other: Path) -> Series: ...
@@ -4101,13 +4099,7 @@ class Series(IndexOpsMixin[S1], ElementOpsMixin[S1], NDFrame):
41014099
@overload
41024100
def rtruediv(
41034101
self: Series[bool] | Series[int] | Series[float],
4104-
other: (
4105-
timedelta
4106-
| Sequence[timedelta]
4107-
| np_ndarray_td
4108-
| TimedeltaIndex
4109-
| Series[Timedelta]
4110-
),
4102+
other: ScalarArrayIndexSeriesTimedelta,
41114103
level: Level | None = None,
41124104
fill_value: float | None = None,
41134105
axis: AxisIndex = 0,

‎tests/series/arithmetic/float/test_floordiv.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def test_floordiv_py_sequence(left: "pd.Series[float]") -> None:
8787
if TYPE_CHECKING_INVALID_USAGE:
8888
_13 = c // left # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
8989
_14 = s // left # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
90-
_15 = d // left # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
90+
check(assert_type(d // left, pd.Series), pd.Series, timedelta)
9191

9292
check(assert_type(left.floordiv(b), "pd.Series[float]"), pd.Series, np.floating)
9393
check(assert_type(left.floordiv(i), "pd.Series[float]"), pd.Series, np.floating)

‎tests/series/arithmetic/int/test_floordiv.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def test_floordiv_py_sequence(left: "pd.Series[int]") -> None:
8787
if TYPE_CHECKING_INVALID_USAGE:
8888
_13 = c // left # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
8989
_14 = s // left # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
90-
_15 = d // left # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
90+
check(assert_type(d // left, pd.Series), pd.Series, timedelta)
9191

9292
check(assert_type(left.floordiv(b), "pd.Series[int]"), pd.Series, np.integer)
9393
check(assert_type(left.floordiv(i), "pd.Series[int]"), pd.Series, np.integer)

‎tests/series/arithmetic/test_floordiv.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def test_floordiv_py_sequence(left_i: pd.Series) -> None:
8787
if TYPE_CHECKING_INVALID_USAGE:
8888
_13 = c // left_i # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
8989
_14 = s // left_i # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
90-
_15 = d // left_i # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
90+
check(assert_type(d // left_i, pd.Series), pd.Series, timedelta)
9191

9292
check(assert_type(left_i.floordiv(b), pd.Series), pd.Series, np.integer)
9393
check(assert_type(left_i.floordiv(i), pd.Series), pd.Series, np.integer)

0 commit comments

Comments
 (0)