def test_merge_operacoes_df_eh_none(self):
ticker = random_ticker()
other_data = [{'ticker': ticker, 'qtd': 100, 'data': date(2019, 10, 11), 'preco': 100, 'aquisicao_via': 'HomeBroker'},
{'ticker': ticker, 'qtd': 100, 'data': date(2019, 10, 12), 'preco': 100, 'aquisicao_via': 'HomeBroker'}]
other_df = create_testing_dataframe(other_data)
with pytest.raises(AssertionError):
merge_operacoes(None, other_df)
def test_merge_operacoes_other_df_vazio(self):
data = [{
'ticker': 'gcgs',
'qtd': 100,
'data': datetime.date(2019, 3, 11),
'preco': 100,
'aquisicao_via': 'HomeBroker'
}, {
'ticker': 'gcgs',
'qtd': 100,
'data': datetime.date(2019, 4, 11),
'preco': 100,
'aquisicao_via': 'HomeBroker'
}, {
'ticker': 'gcgs',
'qtd': -100,
'data': datetime.date(2019, 4, 12),
'preco': 200,
'aquisicao_via': 'HomeBroker'
}]
df = create_testing_dataframe(data)
other_df = create_testing_dataframe([])
result = merge_operacoes(df, other_df)
assert len(result) == len(df)
def do_busca_trades_e_faz_merge_operacoes():
from src.crawler_cei import busca_trades
df_cei = busca_trades(os.environ['CPF'], os.environ['SENHA_CEI'])
from src.dropbox_files import download_dropbox_file
download_dropbox_file()
df = get_operations()
df = merge_operacoes(df, df_cei)
df_to_csv(df, OPERATIONS_FILEPATH)
upload_dropbox_file(OPERATIONS_FILEPATH,
os.environ['DROPBOX_FILE_LOCATION'])
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_merge_operacoes_df_vazio(self):
ticker = random_ticker()
df = create_testing_dataframe([])
other_data = [{'ticker': ticker, 'qtd': 100, 'data': date(2019, 10, 11), 'preco': 100, 'aquisicao_via': 'HomeBroker'},
{'ticker': ticker, 'qtd': 100, 'data': date(2019, 10, 12), 'preco': 100, 'aquisicao_via': 'HomeBroker'}]
other_df = create_testing_dataframe(other_data)
result = merge_operacoes(df, other_df)
assert len(result) == len(other_df)
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(OPERATIONS_FILEPATH)
df = get_operations(filepath=OPERATIONS_FILEPATH)
df = merge_operacoes(df, df_cei)
df_to_csv(df, OPERATIONS_FILEPATH)
upload_dropbox_file(OPERATIONS_FILEPATH, None)
def test_merge_operacoes_com_ipo(self):
data = [{'ticker': 'gcgs', 'qtd': 100, 'data': datetime.date(2019, 3, 11), 'preco': 100, 'aquisicao_via': 'HomeBroker'},
{'ticker': 'gcgs', 'qtd': 100, 'data': datetime.date(2019, 4, 11), 'preco': 100, 'aquisicao_via': 'HomeBroker'}]
other_data = [{'ticker': 'gcgs', 'qtd': 100, 'data': datetime.date(2019, 10, 13), 'preco': 100, 'aquisicao_via': 'HomeBroker'},
{'ticker': 'gcgs', 'qtd': 100, 'data': datetime.date(2019, 10, 14), 'preco': 100, 'aquisicao_via': 'HomeBroker'}]
other_data.extend(data)
other_df = create_testing_dataframe(other_data)
data.extend([{'ticker': 'gcgs', 'qtd': -100, 'data': datetime.date(2019, 4, 20), 'preco': 200, 'aquisicao_via': 'IPO'}])
df = create_testing_dataframe(data)
result = merge_operacoes(df, other_df)
assert len(result) == (len(df) + 2)
def test_merge_operacoes_append_puro(self):
ticker = random_ticker()
data = [{'ticker': ticker, 'qtd': 100, 'data': date(2019, 3, 11), 'preco': 100, 'aquisicao_via': 'HomeBroker'},
{'ticker': ticker, 'qtd': 100, 'data': date(2019, 4, 11), 'preco': 100, 'aquisicao_via': 'HomeBroker'},
{'ticker': ticker, 'qtd': -100, 'data': date(2019, 4, 12), 'preco': 200, 'aquisicao_via': 'HomeBroker'}]
df = create_testing_dataframe(data)
ticker2 = random_ticker()
other_data = [{'ticker': ticker2, 'qtd': 100, 'data': date(2019, 10, 14), 'preco': 100, 'aquisicao_via': 'HomeBroker'},
{'ticker': ticker2, 'qtd': 100, 'data': date(2019, 10, 15), 'preco': 100, 'aquisicao_via': 'HomeBroker'}]
other_df = create_testing_dataframe(other_data)
result = merge_operacoes(df, other_df)
assert len(result) == (len(df) + len(other_df))
