@@ -217,31 +217,31 @@ def _check_generated_range(self, start, freq):
217
217
self .assertEqual (infer_freq (index ), gen .freqstr )
218
218
else :
219
219
inf_freq = infer_freq (index )
220
- self .assert_ ((inf_freq == 'Q-DEC' and
221
- gen .freqstr in ('Q' , 'Q-DEC' , 'Q-SEP' , 'Q-JUN' ,
222
- 'Q-MAR' ))
223
- or
224
- (inf_freq == 'Q-NOV' and
225
- gen .freqstr in ('Q-NOV' , 'Q-AUG' , 'Q-MAY' , 'Q-FEB' ))
226
- or
227
- (inf_freq == 'Q-OCT' and
228
- gen .freqstr in ('Q-OCT' , 'Q-JUL' , 'Q-APR' , 'Q-JAN' )))
220
+ self .assertTrue ((inf_freq == 'Q-DEC' and
221
+ gen .freqstr in ('Q' , 'Q-DEC' , 'Q-SEP' , 'Q-JUN' ,
222
+ 'Q-MAR' ))
223
+ or
224
+ (inf_freq == 'Q-NOV' and
225
+ gen .freqstr in ('Q-NOV' , 'Q-AUG' , 'Q-MAY' , 'Q-FEB' ))
226
+ or
227
+ (inf_freq == 'Q-OCT' and
228
+ gen .freqstr in ('Q-OCT' , 'Q-JUL' , 'Q-APR' , 'Q-JAN' )))
229
229
230
230
gen = date_range (start , periods = 5 , freq = freq )
231
231
index = _dti (gen .values )
232
232
if not freq .startswith ('Q-' ):
233
233
self .assertEqual (infer_freq (index ), gen .freqstr )
234
234
else :
235
235
inf_freq = infer_freq (index )
236
- self .assert_ ((inf_freq == 'Q-DEC' and
237
- gen .freqstr in ('Q' , 'Q-DEC' , 'Q-SEP' , 'Q-JUN' ,
238
- 'Q-MAR' ))
239
- or
240
- (inf_freq == 'Q-NOV' and
241
- gen .freqstr in ('Q-NOV' , 'Q-AUG' , 'Q-MAY' , 'Q-FEB' ))
242
- or
243
- (inf_freq == 'Q-OCT' and
244
- gen .freqstr in ('Q-OCT' , 'Q-JUL' , 'Q-APR' , 'Q-JAN' )))
236
+ self .assertTrue ((inf_freq == 'Q-DEC' and
237
+ gen .freqstr in ('Q' , 'Q-DEC' , 'Q-SEP' , 'Q-JUN' ,
238
+ 'Q-MAR' ))
239
+ or
240
+ (inf_freq == 'Q-NOV' and
241
+ gen .freqstr in ('Q-NOV' , 'Q-AUG' , 'Q-MAY' , 'Q-FEB' ))
242
+ or
243
+ (inf_freq == 'Q-OCT' and
244
+ gen .freqstr in ('Q-OCT' , 'Q-JUL' , 'Q-APR' , 'Q-JAN' )))
245
245
246
246
def test_infer_freq (self ):
247
247
rng = period_range ('1959Q2' , '2009Q3' , freq = 'Q' )
0 commit comments