@@ -141,7 +141,7 @@ def test_sub_pd_series() -> None:
141
141
142
142
143
143
def test_sub_py_datetime () -> None :
144
- """Test pd.Series[Any] - Python native datetime(s) """
144
+ """Test pd.Series[Any] - Python native datetime"""
145
145
s = anchor
146
146
147
147
check (assert_type (left_ts - s , "TimedeltaSeries" ), pd .Series , pd .Timedelta )
@@ -156,22 +156,22 @@ def test_sub_py_datetime() -> None:
156
156
def test_sub_numpy_datetime () -> None :
157
157
"""Test pd.Series[Any] - numpy datetime(s)"""
158
158
s = np .datetime64 (anchor )
159
- a = np .array ([s + np .timedelta64 (m , "m" ) for m in range (3 )])
159
+ a = np .array ([s + np .timedelta64 (m , "m" ) for m in range (3 )], dtype = np . datetime64 )
160
160
161
- # `numpy` typing gives the corresponding `ndarray`s in the static type
162
- # checking, where our `__rsub__` cannot override. At runtime, they return
163
- # `Series`s.
164
161
check (assert_type (left_ts - s , "TimedeltaSeries" ), pd .Series , pd .Timedelta )
165
- check (assert_type (left_ts - a , "TimedeltaSeries" ), pd .Series , pd .Timedelta ) # type: ignore[assert-type]
162
+ check (assert_type (left_ts - a , "TimedeltaSeries" ), pd .Series , pd .Timedelta )
166
163
167
164
check (assert_type (s - left_ts , "TimedeltaSeries" ), pd .Series , pd .Timedelta )
165
+ # `numpy` typing gives the corresponding `ndarray`s in the static type
166
+ # checking, where our `__rsub__` cannot override. At runtime, they return
167
+ # `Series`s.
168
168
check (assert_type (a - left_ts , "npt.NDArray[np.datetime64]" ), pd .Series , pd .Timedelta ) # type: ignore[assert-type]
169
169
170
170
check (assert_type (left_ts .sub (s ), "TimedeltaSeries" ), pd .Series , pd .Timedelta )
171
- check (assert_type (left_ts .sub (a ), "TimedeltaSeries" ), pd .Series , pd .Timedelta ) # type: ignore[assert-type]
171
+ check (assert_type (left_ts .sub (a ), "TimedeltaSeries" ), pd .Series , pd .Timedelta )
172
172
173
173
check (assert_type (left_ts .rsub (s ), "TimedeltaSeries" ), pd .Series , pd .Timedelta )
174
- check (assert_type (left_ts .rsub (a ), "TimedeltaSeries" ), pd .Series , pd .Timedelta ) # type: ignore[assert-type]
174
+ check (assert_type (left_ts .rsub (a ), "TimedeltaSeries" ), pd .Series , pd .Timedelta )
175
175
176
176
177
177
def test_sub_pd_datetime () -> None :
0 commit comments