def preparar_dados_pagamentos(self): self.df_pagamentos = formata.formatar_valor_para_float( self.df_pagamentos) self.df_pagamentos = formata.formatar_data_do_df_string_para_date( self.df_pagamentos) self.df_pagamentos = formata.ordenar_df_data(self.df_pagamentos) self.df_pagamentos = formata.separa_plano_meses_contrato( self.df_pagamentos) self.df_pagamentos = formata.criar_chave_concatenada( self.df_pagamentos)
def cria_data_frame_test(self): obj_pagamentos = { 'ClienteID': [3, 2, 1, 1, 3, 2 ], 'DataPagamento': ['05/03/2017', '12/08/2018', '01/01/2017', '01/02/2017', '05/06/2017', '01/02/2019'], 'Valor': ['R$ 300,00', 'R$ 750,00', 'R$ 399,00', 'R$ 300,00', 'R$ 750,00', 'R$ 750,00' ], 'Plano': ['Bronze/3', 'Ouro/3', 'Platina/1', 'Bronze/3', 'Ouro/3', 'Ouro/3' ] } df = pd.DataFrame(data=obj_pagamentos) df = formata.formatar_valor_para_float(df) df = formata.formatar_data_do_df_string_para_date(df) df = formata.separa_plano_meses_contrato(df) df = formata.criar_chave_concatenada(df) return df
def test_separa_plano_meses_contrato(self): df = self.cria_data_frame_test() df = FormatadorDadosPagamentos.separa_plano_meses_contrato(df) erro = "separa_plano_meses_contrato não separou corretamente" assert df[df['ClienteID'] == 1]['Meses'].iloc[0] == 1, erro assert df[df['ClienteID'] == 1]['PlanoNormalizado'].iloc[0] == 'Platina', erro assert df[df['ClienteID'] == 2]['Meses'].iloc[0] == 3, erro assert df[df['ClienteID'] == 2]['PlanoNormalizado'].iloc[0] == 'Ouro', erro assert df[df['ClienteID'] == 3]['Meses'].iloc[0] == 3, erro assert df[df['ClienteID'] == 3]['PlanoNormalizado'].iloc[0] == 'Bronze', erro