26
26
from pandas ._libs .tslibs .timedeltas import Components
27
27
28
28
from tests import (
29
- IS_TYPE_CHECKER_MYPY ,
30
29
PD_LTE_15 ,
31
30
TYPE_CHECKING_INVALID_USAGE ,
32
31
check ,
@@ -607,14 +606,14 @@ def test_timedelta_add_sub() -> None:
607
606
# TypeError: as_period, as_timestamp, as_datetime, as_date, as_datetime64,
608
607
# as_period_index, as_datetime_index, as_ndarray_dt64
609
608
if TYPE_CHECKING_INVALID_USAGE :
610
- td - as_period # type: ignore[operator]
611
- td - as_timestamp # type: ignore[operator]
612
- td - as_datetime # type: ignore[operator]
613
- td - as_date # type: ignore[operator]
614
- td - as_datetime64 # type: ignore[operator]
615
- td - as_period_index # type: ignore[operator]
616
- td - as_datetime_index # type: ignore[operator]
617
- td - as_ndarray_dt64 # type: ignore[operator]
609
+ td - as_period # type: ignore[operator] # pyright: ignore[reportGeneralTypeIssues]
610
+ td - as_timestamp # type: ignore[operator] # pyright: ignore[reportGeneralTypeIssues]
611
+ td - as_datetime # type: ignore[operator] # pyright: ignore[reportGeneralTypeIssues]
612
+ td - as_date # type: ignore[operator] # pyright: ignore[reportGeneralTypeIssues]
613
+ td - as_datetime64 # type: ignore[operator] # pyright: ignore[reportGeneralTypeIssues]
614
+ td - as_period_index # type: ignore[operator] # pyright: ignore[reportGeneralTypeIssues]
615
+ td - as_datetime_index # type: ignore[operator] # pyright: ignore[reportGeneralTypeIssues]
616
+ td - as_ndarray_dt64 # type: ignore[operator] # pyright: ignore[reportGeneralTypeIssues]
618
617
619
618
check (assert_type (td - td , pd .Timedelta ), pd .Timedelta )
620
619
check (assert_type (td - as_dt_timedelta , pd .Timedelta ), pd .Timedelta )
@@ -746,14 +745,14 @@ def test_timedelta_mul_div() -> None:
746
745
# TypeError: md_int, md_float, md_ndarray_intp, md_ndarray_float, mp_series_int,
747
746
# mp_series_float, md_int64_index, md_float_index
748
747
if TYPE_CHECKING_INVALID_USAGE :
749
- md_int // td # type: ignore[operator]
750
- md_float // td # type: ignore[operator]
751
- md_ndarray_intp // td # type: ignore[operator]
752
- md_ndarray_float // td # type: ignore[operator]
753
- mp_series_int // td # type: ignore[operator]
754
- md_series_float // td # type: ignore[operator]
755
- md_int64_index // td # type: ignore[operator]
756
- md_float_index // td # type: ignore[operator]
748
+ md_int // td # type: ignore[operator] # pyright: ignore[reportGeneralTypeIssues]
749
+ md_float // td # type: ignore[operator] # pyright: ignore[reportGeneralTypeIssues]
750
+ md_ndarray_intp // td # type: ignore[operator] # pyright: ignore[reportGeneralTypeIssues]
751
+ md_ndarray_float // td # type: ignore[operator] # pyright: ignore[reportGeneralTypeIssues]
752
+ mp_series_int // td # type: ignore[operator] # pyright: ignore[reportGeneralTypeIssues]
753
+ md_series_float // td # type: ignore[operator] # pyright: ignore[reportGeneralTypeIssues]
754
+ md_int64_index // td # type: ignore[operator] # pyright: ignore[reportGeneralTypeIssues]
755
+ md_float_index // td # type: ignore[operator] # pyright: ignore[reportGeneralTypeIssues]
757
756
758
757
check (assert_type (td / td , float ), float )
759
758
check (assert_type (td / pd .NaT , float ), float )
@@ -780,14 +779,14 @@ def test_timedelta_mul_div() -> None:
780
779
# TypeError: md_int, md_float, md_ndarray_intp, md_ndarray_float, mp_series_int,
781
780
# mp_series_float, md_int64_index, md_float_index
782
781
if TYPE_CHECKING_INVALID_USAGE :
783
- md_int / td # type: ignore[operator]
784
- md_float / td # type: ignore[operator]
785
- md_ndarray_intp / td # type: ignore[operator]
786
- md_ndarray_float / td # type: ignore[operator]
787
- mp_series_int / td # type: ignore[operator]
788
- md_series_float / td # type: ignore[operator]
789
- md_int64_index / td # type: ignore[operator]
790
- md_float_index / td # type: ignore[operator]
782
+ md_int / td # type: ignore[operator] # pyright: ignore[reportGeneralTypeIssues]
783
+ md_float / td # type: ignore[operator] # pyright: ignore[reportGeneralTypeIssues]
784
+ md_ndarray_intp / td # type: ignore[operator] # pyright: ignore[reportGeneralTypeIssues]
785
+ md_ndarray_float / td # type: ignore[operator] # pyright: ignore[reportGeneralTypeIssues]
786
+ mp_series_int / td # type: ignore[operator] # pyright: ignore[reportGeneralTypeIssues]
787
+ md_series_float / td # type: ignore[operator] # pyright: ignore[reportGeneralTypeIssues]
788
+ md_int64_index / td # type: ignore[operator] # pyright: ignore[reportGeneralTypeIssues]
789
+ md_float_index / td # type: ignore[operator] # pyright: ignore[reportGeneralTypeIssues]
791
790
792
791
793
792
def test_timedelta_mod_abs_unary () -> None :
@@ -826,12 +825,9 @@ def test_timedelta_mod_abs_unary() -> None:
826
825
pd .TimedeltaIndex ,
827
826
)
828
827
829
- if TYPE_CHECKING and IS_TYPE_CHECKER_MYPY :
830
- # mypy reports dt.timedelta, even though __abs__ returns Timedelta
831
- check (assert_type (abs (td ), pd .Timedelta ), pd .Timedelta ) # type: ignore[assert-type]
832
- else :
833
- # This is valid for pyright
834
- check (assert_type (abs (td ), pd .Timedelta ), pd .Timedelta )
828
+ # mypy reports dt.timedelta, even though __abs__ returns Timedelta
829
+ check (assert_type (abs (td ), pd .Timedelta ), pd .Timedelta ) # type: ignore[assert-type]
830
+
835
831
check (assert_type (td .__abs__ (), pd .Timedelta ), pd .Timedelta )
836
832
check (assert_type (- td , pd .Timedelta ), pd .Timedelta )
837
833
check (assert_type (+ td , pd .Timedelta ), pd .Timedelta )
0 commit comments