2424# left operands
2525left_i = pd .DataFrame ({"a" : [1 , 2 , 3 ]})["a" ]
2626left_ts = pd .DataFrame ({"a" : [anchor + timedelta (hours = h + 1 ) for h in range (3 )]})["a" ]
27- left_td = pd .DataFrame ({"a" : [timedelta (hours = h , minutes = 1 ) for h in range (3 )]})["a" ]
2827
2928
3029def test_sub_i_py_scalar () -> None :
@@ -155,48 +154,34 @@ def test_sub_i_pd_series() -> None:
155154
156155
157156def test_sub_ts_py_datetime () -> None :
158- """Test pd.Series[Any] (Timestamp | Timedelta ) - Python native datetime"""
157+ """Test pd.Series[Any] (Timestamp) - Python native datetime"""
159158 s = anchor
160159 a = [s + timedelta (minutes = m ) for m in range (3 )]
161160
162161 check (assert_type (left_ts - s , "TimedeltaSeries" ), pd .Series , pd .Timedelta )
163162 if TYPE_CHECKING_INVALID_USAGE :
164163 _1 = left_ts - a # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
165- check (assert_type (left_td - s , "TimedeltaSeries" ), pd .Series , pd .Timedelta )
166- if TYPE_CHECKING_INVALID_USAGE :
167- _3 = left_td - a # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
168164
169165 check (assert_type (s - left_ts , "TimedeltaSeries" ), pd .Series , pd .Timedelta )
170166 if TYPE_CHECKING_INVALID_USAGE :
171167 _5 = a - left_ts # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
172- check (assert_type (s - left_td , "TimedeltaSeries" ), pd .Series , pd .Timedelta )
173- if TYPE_CHECKING_INVALID_USAGE :
174- _7 = a - left_td # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
175168
176169 check (assert_type (left_ts .sub (s ), "TimedeltaSeries" ), pd .Series , pd .Timedelta )
177170 if TYPE_CHECKING_INVALID_USAGE :
178171 left_ts .sub (a ) # type: ignore[arg-type] # pyright: ignore[reportArgumentType,reportCallIssue]
179- check (assert_type (left_td .sub (s ), "TimedeltaSeries" ), pd .Series , pd .Timedelta )
180- if TYPE_CHECKING_INVALID_USAGE :
181- left_td .sub (a ) # type: ignore[arg-type] # pyright: ignore[reportArgumentType,reportCallIssue]
182172
183- check (assert_type (left_td .rsub (s ), "TimedeltaSeries" ), pd .Series , pd .Timedelta )
173+ check (assert_type (left_ts .rsub (s ), "TimedeltaSeries" ), pd .Series , pd .Timedelta )
184174 if TYPE_CHECKING_INVALID_USAGE :
185175 left_ts .rsub (a ) # type: ignore[arg-type] # pyright: ignore[reportArgumentType,reportCallIssue]
186- check (assert_type (left_td .rsub (s ), "TimedeltaSeries" ), pd .Series , pd .Timedelta )
187- if TYPE_CHECKING_INVALID_USAGE :
188- left_td .rsub (a ) # type: ignore[arg-type] # pyright: ignore[reportArgumentType,reportCallIssue]
189176
190177
191178def test_sub_ts_numpy_datetime () -> None :
192- """Test pd.Series[Any] (Timestamp | Timedelta ) - numpy datetime(s)"""
179+ """Test pd.Series[Any] (Timestamp) - numpy datetime(s)"""
193180 s = np .datetime64 (anchor )
194181 a = np .array ([s + np .timedelta64 (m , "m" ) for m in range (3 )], np .datetime64 )
195182
196183 check (assert_type (left_ts - s , "TimedeltaSeries" ), pd .Series , pd .Timedelta )
197184 check (assert_type (left_ts - a , "TimedeltaSeries" ), pd .Series , pd .Timedelta )
198- check (assert_type (left_td - s , "TimedeltaSeries" ), pd .Series , pd .Timedelta )
199- check (assert_type (left_td - a , "TimedeltaSeries" ), pd .Series , pd .Timedelta )
200185
201186 # `numpy` typing gives the corresponding `ndarray`s in the static type
202187 # checking, where our `__rsub__` cannot override. At runtime, they return
@@ -208,53 +193,32 @@ def test_sub_ts_numpy_datetime() -> None:
208193 pd .Series ,
209194 pd .Timedelta ,
210195 )
211- check (assert_type (s - left_td , "TimedeltaSeries" ), pd .Series , pd .Timedelta )
212- check (
213- assert_type (a - left_td , Any ), # pyright: ignore[reportAssertTypeFailure]
214- pd .Series ,
215- pd .Timedelta ,
216- )
217196
218197 check (assert_type (left_ts .sub (s ), "TimedeltaSeries" ), pd .Series , pd .Timedelta )
219198 check (assert_type (left_ts .sub (a ), "TimedeltaSeries" ), pd .Series , pd .Timedelta )
220- check (assert_type (left_td .sub (s ), "TimedeltaSeries" ), pd .Series , pd .Timedelta )
221- check (assert_type (left_td .sub (a ), "TimedeltaSeries" ), pd .Series , pd .Timedelta )
222199
223200 check (assert_type (left_ts .rsub (s ), "TimedeltaSeries" ), pd .Series , pd .Timedelta )
224201 check (assert_type (left_ts .rsub (a ), "TimedeltaSeries" ), pd .Series , pd .Timedelta )
225- check (assert_type (left_td .rsub (s ), "TimedeltaSeries" ), pd .Series , pd .Timedelta )
226- check (assert_type (left_td .rsub (a ), "TimedeltaSeries" ), pd .Series , pd .Timedelta )
227202
228203
229204def test_sub_ts_pd_datetime () -> None :
230- """Test pd.Series[Any] (Timestamp | Timedelta ) - Pandas datetime(s)"""
205+ """Test pd.Series[Any] (Timestamp) - Pandas datetime(s)"""
231206 s = pd .Timestamp (anchor )
232207 a = pd .Series ([s + pd .Timedelta (minutes = m ) for m in range (3 )])
233208
234209 check (assert_type (left_ts - s , "TimedeltaSeries" ), pd .Series , pd .Timedelta )
235210 check (assert_type (left_ts - a , "pd.Series[pd.Timedelta]" ), pd .Series , pd .Timedelta )
236- check (assert_type (left_td - s , "TimedeltaSeries" ), pd .Series , pd .Timedelta )
237- check (assert_type (left_td - a , "pd.Series[pd.Timedelta]" ), pd .Series , pd .Timedelta )
238211
239- left_ts .__rsub__ (s )
240212 check (assert_type (s - left_ts , "TimedeltaSeries" ), pd .Series , pd .Timedelta )
241213 check (assert_type (a - left_ts , pd .Series ), pd .Series , pd .Timedelta )
242- check (assert_type (s - left_td , "TimedeltaSeries" ), pd .Series , pd .Timedelta )
243- check (assert_type (a - left_td , pd .Series ), pd .Series , pd .Timedelta )
244214
245215 check (assert_type (left_ts .sub (s ), "TimedeltaSeries" ), pd .Series , pd .Timedelta )
246216 check (
247217 assert_type (left_ts .sub (a ), "pd.Series[pd.Timedelta]" ), pd .Series , pd .Timedelta
248218 )
249- check (assert_type (left_td .sub (s ), "TimedeltaSeries" ), pd .Series , pd .Timedelta )
250- check (
251- assert_type (left_td .sub (a ), "pd.Series[pd.Timedelta]" ), pd .Series , pd .Timedelta
252- )
253219
254220 check (assert_type (left_ts .rsub (s ), "TimedeltaSeries" ), pd .Series , pd .Timedelta )
255221 check (assert_type (left_ts .rsub (a ), pd .Series ), pd .Series , pd .Timedelta )
256- check (assert_type (left_td .rsub (s ), "TimedeltaSeries" ), pd .Series , pd .Timedelta )
257- check (assert_type (left_td .rsub (a ), pd .Series ), pd .Series , pd .Timedelta )
258222
259223
260224def test_sub_str_py_str () -> None :
0 commit comments