Skip to content

Commit 27e81a0

Browse files
cmp0xffcmp0xff
committed
1 parent 9e840fc commit 27e81a0

File tree

1 file changed

+60
-19
lines changed

1 file changed

+60
-19
lines changed

‎tests/series/arithmetic/timedelta/test_mul.py‎

Lines changed: 60 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -19,25 +19,36 @@ def test_mul_py_scalar() -> None:
1919
"""Test pd.Series[pd.Timedelta] * Python native scalars"""
2020
b, i, f, c = True, 1, 1.0, 1j
2121

22-
check(assert_type(left * b, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
22+
# pandas-dev/pandas#62316
23+
if PD_LTE_23:
24+
check(assert_type(left * b, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
2325
check(assert_type(left * i, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
2426
check(assert_type(left * f, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
2527
if TYPE_CHECKING_INVALID_USAGE:
2628
_0 = left * c # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
2729

28-
check(assert_type(b * left, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
30+
if PD_LTE_23:
31+
check(assert_type(b * left, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
2932
check(assert_type(i * left, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
3033
check(assert_type(f * left, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
3134
if TYPE_CHECKING_INVALID_USAGE:
3235
_1 = c * left # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
3336

34-
check(assert_type(left.mul(b), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
37+
if PD_LTE_23:
38+
check(
39+
assert_type(left.mul(b), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta
40+
)
3541
check(assert_type(left.mul(i), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
3642
check(assert_type(left.mul(f), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
3743
if TYPE_CHECKING_INVALID_USAGE:
3844
left.mul(c) # type: ignore[call-overload] # pyright: ignore[reportArgumentType,reportCallIssue]
3945

40-
check(assert_type(left.rmul(b), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
46+
if PD_LTE_23:
47+
check(
48+
assert_type(left.rmul(b), "pd.Series[pd.Timedelta]"),
49+
pd.Series,
50+
pd.Timedelta,
51+
)
4152
check(assert_type(left.rmul(i), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
4253
check(assert_type(left.rmul(f), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
4354
if TYPE_CHECKING_INVALID_USAGE:
@@ -48,25 +59,36 @@ def test_mul_py_sequence() -> None:
4859
"""Test pd.Series[pd.Timedelta] * Python native sequences"""
4960
b, i, f, c = [True], [2], [1.5], [1.7j]
5061

51-
check(assert_type(left * b, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
62+
# pandas-dev/pandas#62316
63+
if PD_LTE_23:
64+
check(assert_type(left * b, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
5265
check(assert_type(left * i, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
5366
check(assert_type(left * f, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
5467
if TYPE_CHECKING_INVALID_USAGE:
5568
_0 = left * c # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
5669

57-
check(assert_type(b * left, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
70+
if PD_LTE_23:
71+
check(assert_type(b * left, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
5872
check(assert_type(i * left, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
5973
check(assert_type(f * left, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
6074
if TYPE_CHECKING_INVALID_USAGE:
6175
_1 = c * left # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
6276

63-
check(assert_type(left.mul(b), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
77+
if PD_LTE_23:
78+
check(
79+
assert_type(left.mul(b), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta
80+
)
6481
check(assert_type(left.mul(i), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
6582
check(assert_type(left.mul(f), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
6683
if TYPE_CHECKING_INVALID_USAGE:
6784
left.mul(c) # type: ignore[arg-type] # pyright: ignore[reportArgumentType,reportCallIssue]
6885

69-
check(assert_type(left.rmul(b), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
86+
if PD_LTE_23:
87+
check(
88+
assert_type(left.rmul(b), "pd.Series[pd.Timedelta]"),
89+
pd.Series,
90+
pd.Timedelta,
91+
)
7092
check(assert_type(left.rmul(i), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
7193
check(assert_type(left.rmul(f), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
7294
if TYPE_CHECKING_INVALID_USAGE:
@@ -80,7 +102,9 @@ def test_mul_numpy_array() -> None:
80102
f = np.array([1.5], np.float64)
81103
c = np.array([1.7j], np.complex128)
82104

83-
check(assert_type(left * b, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
105+
# pandas-dev/pandas#62316
106+
if PD_LTE_23:
107+
check(assert_type(left * b, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
84108
check(assert_type(left * i, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
85109
check(assert_type(left * f, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
86110
if TYPE_CHECKING_INVALID_USAGE:
@@ -89,20 +113,29 @@ def test_mul_numpy_array() -> None:
89113
# `numpy` typing gives the corresponding `ndarray`s in the static type
90114
# checking, where our `__rmul__` cannot override. At runtime, they return
91115
# `Series` with the correct element type.
92-
check(assert_type(b * left, "npt.NDArray[np.bool_]"), pd.Series, pd.Timedelta)
116+
if PD_LTE_23:
117+
check(assert_type(b * left, "npt.NDArray[np.bool_]"), pd.Series, pd.Timedelta)
93118
check(assert_type(i * left, "npt.NDArray[np.int64]"), pd.Series, pd.Timedelta)
94119
check(assert_type(f * left, "npt.NDArray[np.float64]"), pd.Series, pd.Timedelta)
95120
if TYPE_CHECKING_INVALID_USAGE:
96121
# We made it Never, but numpy takes over
97122
assert_type(c * left, "npt.NDArray[np.complex128]")
98123

99-
check(assert_type(left.mul(b), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
124+
if PD_LTE_23:
125+
check(
126+
assert_type(left.mul(b), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta
127+
)
100128
check(assert_type(left.mul(i), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
101129
check(assert_type(left.mul(f), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
102130
if TYPE_CHECKING_INVALID_USAGE:
103131
left.mul(c) # type: ignore[arg-type] # pyright: ignore[reportArgumentType,reportCallIssue]
104132

105-
check(assert_type(left.rmul(b), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
133+
if PD_LTE_23:
134+
check(
135+
assert_type(left.rmul(b), "pd.Series[pd.Timedelta]"),
136+
pd.Series,
137+
pd.Timedelta,
138+
)
106139
check(assert_type(left.rmul(i), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
107140
check(assert_type(left.rmul(f), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
108141
if TYPE_CHECKING_INVALID_USAGE:
@@ -116,25 +149,36 @@ def test_mul_pd_index() -> None:
116149
f = pd.Index([1.5])
117150
c = pd.Index([1.7j])
118151

119-
check(assert_type(left * b, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
152+
# pandas-dev/pandas#62316
153+
if PD_LTE_23:
154+
check(assert_type(left * b, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
120155
check(assert_type(left * i, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
121156
check(assert_type(left * f, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
122157
if TYPE_CHECKING_INVALID_USAGE:
123158
_0 = left * c # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
124159

125-
check(assert_type(b * left, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
160+
if PD_LTE_23:
161+
check(assert_type(b * left, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
126162
check(assert_type(i * left, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
127163
check(assert_type(f * left, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
128164
if TYPE_CHECKING_INVALID_USAGE:
129165
_1 = c * left # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
130166

131-
check(assert_type(left.mul(b), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
167+
if PD_LTE_23:
168+
check(
169+
assert_type(left.mul(b), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta
170+
)
132171
check(assert_type(left.mul(i), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
133172
check(assert_type(left.mul(f), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
134173
if TYPE_CHECKING_INVALID_USAGE:
135174
left.mul(c) # type: ignore[arg-type] # pyright: ignore[reportArgumentType,reportCallIssue]
136175

137-
check(assert_type(left.rmul(b), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
176+
if PD_LTE_23:
177+
check(
178+
assert_type(left.rmul(b), "pd.Series[pd.Timedelta]"),
179+
pd.Series,
180+
pd.Timedelta,
181+
)
138182
check(assert_type(left.rmul(i), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
139183
check(assert_type(left.rmul(f), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
140184
if TYPE_CHECKING_INVALID_USAGE:
@@ -156,15 +200,13 @@ def test_mul_pd_series() -> None:
156200
if TYPE_CHECKING_INVALID_USAGE:
157201
_0 = left * c # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
158202

159-
# pandas-dev/pandas#62316
160203
if PD_LTE_23:
161204
check(assert_type(b * left, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
162205
check(assert_type(i * left, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
163206
check(assert_type(f * left, "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta)
164207
if TYPE_CHECKING_INVALID_USAGE:
165208
_1 = c * left # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
166209

167-
# pandas-dev/pandas#62316
168210
if PD_LTE_23:
169211
check(
170212
assert_type(left.mul(b), "pd.Series[pd.Timedelta]"), pd.Series, pd.Timedelta
@@ -174,7 +216,6 @@ def test_mul_pd_series() -> None:
174216
if TYPE_CHECKING_INVALID_USAGE:
175217
left.mul(c) # type: ignore[arg-type] # pyright: ignore[reportArgumentType,reportCallIssue]
176218

177-
# pandas-dev/pandas#62316
178219
if PD_LTE_23:
179220
check(
180221
assert_type(left.rmul(b), "pd.Series[pd.Timedelta]"),

0 commit comments

Comments
 (0)