@@ -117,9 +117,8 @@ def test_period() -> None:
117
117
check (assert_type (as4 + p , pd .Period ), pd .Period )
118
118
check (assert_type (as7 + p , "PeriodSeries" ), pd .Series )
119
119
check (assert_type (p .freq + p , pd .Period ), pd .Period )
120
- # TODO: PeriodIndex should have a __sub__ with correct types, this op is valid
121
- # and so the assert_type is skipped
122
- check (as5 - p , pd .Index ) # type: ignore[operator]
120
+
121
+ check (assert_type (as5 - p , pd .Index ), pd .Index )
123
122
124
123
check (assert_type (p .__radd__ (as0 ), pd .Period ), pd .Period )
125
124
check (assert_type (p .__radd__ (as1 ), pd .Period ), pd .Period )
@@ -164,10 +163,15 @@ def test_period() -> None:
164
163
p3 = pd .Period ("2007-01" , freq = "M" )
165
164
check (assert_type (p3 .to_timestamp ("D" , "S" ), pd .Timestamp ), pd .Timestamp )
166
165
check (assert_type (p3 .to_timestamp ("D" , "E" ), pd .Timestamp ), pd .Timestamp )
166
+ check (assert_type (p3 .to_timestamp ("D" , "start" ), pd .Timestamp ), pd .Timestamp )
167
+ check (assert_type (p3 .to_timestamp ("D" , "end" ), pd .Timestamp ), pd .Timestamp )
167
168
check (assert_type (p3 .to_timestamp ("D" , "Finish" ), pd .Timestamp ), pd .Timestamp )
168
- check (assert_type (p3 .to_timestamp ("D" , "End" ), pd .Timestamp ), pd .Timestamp )
169
169
check (assert_type (p3 .to_timestamp ("D" , "Begin" ), pd .Timestamp ), pd .Timestamp )
170
- check (assert_type (p3 .to_timestamp ("D" , "Start" ), pd .Timestamp ), pd .Timestamp )
170
+ check (assert_type (p3 .to_timestamp ("D" , "End" ), pd .Timestamp ), pd .Timestamp )
171
+ check (assert_type (p3 .to_timestamp ("D" , "s" ), pd .Timestamp ), pd .Timestamp )
172
+ check (assert_type (p3 .to_timestamp ("D" , "e" ), pd .Timestamp ), pd .Timestamp )
173
+ check (assert_type (p3 .to_timestamp ("D" , "finish" ), pd .Timestamp ), pd .Timestamp )
174
+ check (assert_type (p3 .to_timestamp ("D" , "begin" ), pd .Timestamp ), pd .Timestamp )
171
175
172
176
check (assert_type (p3 .asfreq ("D" , "S" ), pd .Period ), pd .Period )
173
177
check (assert_type (p3 .asfreq (Day (), "E" ), pd .Period ), pd .Period )
0 commit comments