|
3 | 3 | import numpy as np
|
4 | 4 | from numpy import typing as npt # noqa: F401
|
5 | 5 | import pandas as pd
|
| 6 | +from pandas._testing import ensure_clean |
6 | 7 | from typing_extensions import assert_type
|
7 | 8 |
|
8 | 9 | from tests import (
|
@@ -145,25 +146,25 @@ def test_truediv_pd_series() -> None:
|
145 | 146 |
|
146 | 147 | def test_path_div() -> None:
|
147 | 148 | # GH 682
|
148 |
| - folder = Path.cwd() |
149 |
| - |
150 |
| - folders = pd.Series([folder, folder]) |
151 |
| - check(assert_type(folders / Path("a.png"), pd.Series), pd.Series, Path) |
| 149 | + with ensure_clean() as folder: |
| 150 | + folders = pd.Series([folder, folder]) |
| 151 | + check(assert_type(folders / Path("a.png"), pd.Series), pd.Series, Path) |
152 | 152 |
|
153 | 153 |
|
154 | 154 | def test_truediv_path() -> None:
|
155 | 155 | """Test pd.Series / path object.
|
156 | 156 |
|
157 | 157 | Also GH 682."""
|
158 |
| - left, p = pd.Series(["a.png", "b.gz", "c.txt"]), Path.cwd() |
| 158 | + left = pd.Series(["a.png", "b.gz", "c.txt"]) |
159 | 159 |
|
160 |
| - check(assert_type(left / p, pd.Series), pd.Series, Path) |
161 |
| - if PD_LTE_23: |
162 |
| - # Bug in 3.0 https://github.com/pandas-dev/pandas/issues/61940 |
163 |
| - check(assert_type(p / left, pd.Series), pd.Series, Path) |
| 160 | + with ensure_clean as p: |
| 161 | + check(assert_type(left / p, pd.Series), pd.Series, Path) |
| 162 | + if PD_LTE_23: |
| 163 | + # Bug in 3.0 https://github.com/pandas-dev/pandas/issues/61940 |
| 164 | + check(assert_type(p / left, pd.Series), pd.Series, Path) |
164 | 165 |
|
165 |
| - check(assert_type(left.truediv(p), pd.Series), pd.Series, Path) |
166 |
| - check(assert_type(left.div(p), pd.Series), pd.Series, Path) |
| 166 | + check(assert_type(left.truediv(p), pd.Series), pd.Series, Path) |
| 167 | + check(assert_type(left.div(p), pd.Series), pd.Series, Path) |
167 | 168 |
|
168 |
| - check(assert_type(left.rtruediv(p), pd.Series), pd.Series, Path) |
169 |
| - check(assert_type(left.rdiv(p), pd.Series), pd.Series, Path) |
| 169 | + check(assert_type(left.rtruediv(p), pd.Series), pd.Series, Path) |
| 170 | + check(assert_type(left.rdiv(p), pd.Series), pd.Series, Path) |
0 commit comments