@@ -217,31 +217,31 @@ def _check_generated_range(self, start, freq):
217217 self .assertEqual (infer_freq (index ), gen .freqstr )
218218 else :
219219 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' )))
229229
230230 gen = date_range (start , periods = 5 , freq = freq )
231231 index = _dti (gen .values )
232232 if not freq .startswith ('Q-' ):
233233 self .assertEqual (infer_freq (index ), gen .freqstr )
234234 else :
235235 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' )))
245245
246246 def test_infer_freq (self ):
247247 rng = period_range ('1959Q2' , '2009Q3' , freq = 'Q' )
0 commit comments