Skip to content

Commit 2668636

Browse files
committed
support Path div
1 parent 42ab31a commit 2668636

File tree

3 files changed

+20
-0
lines changed

3 files changed

+20
-0
lines changed

pandas-stubs/core/series.pyi

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@ from datetime import (
1313
time,
1414
timedelta,
1515
)
16+
from pathlib import Path
1617
from typing import (
1718
Any,
1819
ClassVar,
@@ -1649,6 +1650,9 @@ class Series(IndexOpsMixin[S1], NDFrame):
16491650
self, other: int | np_ndarray_anyint | Series[int]
16501651
) -> Series[int]: ...
16511652
def __rsub__(self, other: num | _ListLike | Series[S1]) -> Series: ...
1653+
@overload
1654+
def __rtruediv__(self, other: Path) -> Series: ...
1655+
@overload
16521656
def __rtruediv__(self, other: num | _ListLike | Series[S1]) -> Series: ...
16531657
# ignore needed for mypy as we want different results based on the arguments
16541658
@overload # type: ignore[override]
@@ -1675,6 +1679,9 @@ class Series(IndexOpsMixin[S1], NDFrame):
16751679
) -> TimedeltaSeries: ...
16761680
@overload
16771681
def __sub__(self, other: num | _ListLike | Series) -> Series: ...
1682+
@overload
1683+
def __truediv__(self, other: Path) -> Series: ...
1684+
@overload
16781685
def __truediv__(self, other: num | _ListLike | Series[S1]) -> Series: ...
16791686
# ignore needed for mypy as we want different results based on the arguments
16801687
@overload # type: ignore[override]

tests/test_frame.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3784,6 +3784,8 @@ def test_frame_index_timestamp() -> None:
37843784

37853785

37863786
def test_frame_bool_fails() -> None:
3787+
# GH 663
3788+
37873789
df = pd.DataFrame({"a": [1, 2, 3], "b": [4, 5, 6]})
37883790

37893791
try:

tests/test_series.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3508,6 +3508,7 @@ def test_series_index_timestamp() -> None:
35083508

35093509

35103510
def test_series_bool_fails() -> None:
3511+
# GH 663
35113512
s = pd.Series([1, 2, 3])
35123513

35133514
try:
@@ -3518,3 +3519,13 @@ def test_series_bool_fails() -> None:
35183519
a = s[0]
35193520
except ValueError:
35203521
pass
3522+
3523+
3524+
def test_path_div() -> None:
3525+
# GH 682
3526+
folder = Path.cwd()
3527+
files = pd.Series(["a.png", "b.png"])
3528+
check(assert_type(folder / files, pd.Series), pd.Series, Path)
3529+
3530+
folders = pd.Series([folder, folder])
3531+
check(assert_type(folders / Path("a.png"), pd.Series), pd.Series, Path)

0 commit comments

Comments
 (0)