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_formatar_data_do_df_string_para_date_verifica_tipo(self):
df = self.cria_data_frame_test()
df = FormatadorDadosPagamentos.formatar_data_do_df_string_para_date(df)
assert df[
"DataPagamento"].dtype == 'datetime64[ns]', "DataPagamento deve ser formatado para 'datetime64[ns]'"