def test_multiple_aggr(self):
''' Verifica se retorna np.mean se agregação nula '''
model = BaseModel()
aggrs = ['sum', 'count', 'xpto', '']
vals = ['vl_indicador', 'cd_municipio', 'nu_competencia',
'another_field']
expected = {
'vl_indicador': np.sum,
'cd_municipio': np.sum,
'nu_competencia': np.mean,
'another_field': np.mean
}
self.assertEqual(model.convert_aggr_to_np(aggrs, vals), expected)
def test_single_aggr(self):
''' Verifica se retorna np.mean se agregação nula '''
model = BaseModel()
aggrs = 'sum'
self.assertEqual(model.convert_aggr_to_np(aggrs, None), np.sum)
