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))