def test_df_to_csv(self):
df_original = get_operations_dataframe()
df_to_csv(df_original, 'df_to_csv_testing.txt')
df_lido = get_operations_dataframe('df_to_csv_testing.txt')
assert df_lido.equals(df_original)
def test_calcula_precos_medios_do_dropbox(self):
from src.dropbox_files import download_dropbox_file
download_dropbox_file()
df = get_operations_dataframe()
precos_medios_de_compra = calcula_precos_medio_de_compra(df)
assert type(precos_medios_de_compra) is dict
def do_busca_trades_e_faz_merge_operacoes():
from src.crawler_cei import CrawlerCei
crawler_cei = CrawlerCei(headless=True)
df_cei = crawler_cei.busca_trades()
from src.dropbox_files import download_dropbox_file
download_dropbox_file()
df = get_operations_dataframe()
df = merge_operacoes(df, df_cei)
df_to_csv(df, OPERATIONS_FILEPATH)
upload_dropbox_file(OPERATIONS_FILEPATH, os.environ['DROPBOX_FILE_LOCATION'])
def test_relatorio_txt(self):
from src.dropbox_files import download_dropbox_file
download_dropbox_file()
df = get_operations_dataframe()
df = df.tail(150)
calcula_custodia(df)
calculo_ir = CalculoIr(df=df)
calculo_ir.calcula()
with open("relatorio.txt", "w") as relatorio:
relatorio.write(relatorio_txt(calculo_ir))
def test_relatorio_html(self):
from src.dropbox_files import download_dropbox_file
download_dropbox_file()
df = get_operations_dataframe()
df = df.tail(150)
calcula_custodia(df)
calculo_ir = CalculoIr(df=df)
calculo_ir.calcula()
from py_w3c.validators.html.validator import HTMLValidator
assert HTMLValidator().validate_fragment(relatorio_html(calculo_ir))
def do_calculo_ir():
from src.dropbox_files import download_dropbox_file
download_dropbox_file()
df = get_operations_dataframe()
from src.stuff import calcula_custodia
calcula_custodia(df)
calculo_ir = CalculoIr(df=df)
calculo_ir.calcula()
print(relatorio_txt(calculo_ir))
envia_relatorio_html_por_email(assunto(calculo_ir),
relatorio_html(calculo_ir))
def do_calculo_ir():
from src.dropbox_files import download_dropbox_file
download_dropbox_file()
df = get_operations_dataframe()
from src.stuff import calcula_custodia
calcula_custodia(df)
calculo_ir = CalculoIr(df=df)
calculo_ir.calcula()
print(relatorio_txt(calculo_ir))
envia_relatorio_html_por_email(
'Calculo de IR - ' + calculo_ir.mes_do_relatorio + ' - CPF: ' +
os.environ['CPF'] + ' - ' + datetime.now().strftime("%H:%M:%S"),
relatorio_html(calculo_ir))
def test_calcula_custodia(self):
from src.dropbox_files import download_dropbox_file
download_dropbox_file()
df = get_operations_dataframe()
data = datetime.datetime.now().date()
custodia = calcula_custodia(df, data)
assert type(custodia) is pd.DataFrame
assert 'preco_medio_compra' in custodia.columns
assert 'ticker' in custodia.columns
assert 'qtd' in custodia.columns
assert 'preco_atual' in custodia.columns
assert 'valorizacao' in custodia.columns
assert 'valor' in custodia.columns
assert 'tipo' in custodia.columns
assert 'data_primeira_compra' in custodia.columns
def test_deve_criar_dataframe_vazio_se_arquivo_vazio(self):
df = get_operations_dataframe('arquivo_invalido.txt')
self.assertListEqual(list(df.columns), todas_as_colunas())
