@@ -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,23 +163,20 @@ 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 )
174
- check (assert_type (p3 .asfreq (Day (), "Finish" ), pd .Period ), pd .Period )
175
- check (assert_type (p3 .asfreq (Day (), "Begin" ), pd .Period ), pd .Period )
176
- check (assert_type (p3 .asfreq (Day (), "Start" ), pd .Period ), pd .Period )
177
- check (assert_type (p3 .asfreq (Day (), "End" ), pd .Period ), pd .Period )
178
178
check (assert_type (p3 .asfreq (Day (), "end" ), pd .Period ), pd .Period )
179
179
check (assert_type (p3 .asfreq (Day (), "start" ), pd .Period ), pd .Period )
180
- check (assert_type (p3 .asfreq (Day (), "begin" ), pd .Period ), pd .Period )
181
- check (assert_type (p3 .asfreq (Day (), "finish" ), pd .Period ), pd .Period )
182
- check (assert_type (p3 .asfreq (Day (), "s" ), pd .Period ), pd .Period )
183
- check (assert_type (p3 .asfreq (Day (), "e" ), pd .Period ), pd .Period )
184
180
185
181
check (assert_type (pd .Period .now ("D" ), pd .Period ), pd .Period )
186
182
check (assert_type (pd .Period .now (Day ()), pd .Period ), pd .Period )
0 commit comments