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