def test_default_on_none(self):
''' Verifica se retorna np.mean se agregação nula '''
self.assertEqual(BaseModel.aggr_to_np(None), np.mean)
def test_sum_on_count(self):
''' Verifica se não retorna nada e a agregação for count '''
self.assertEqual(BaseModel.aggr_to_np('count'), np.sum)
def test_default_on_unmapped(self):
''' Verifica se retorna np.mean se agregação não prevista '''
self.assertEqual(BaseModel.aggr_to_np('xpto'), np.mean)
def test_sum(self):
''' Verifica se retorna np.sum '''
self.assertEqual(BaseModel.aggr_to_np('sum'), np.sum)
def test_default_on_empty(self):
''' Verifica se retorna np.mean se agragação vazia '''
self.assertEqual(BaseModel.aggr_to_np(''), np.mean)
