portfolio_management / test_background_task.py
huggingface112's picture
move files to normal tracking except .db
976166f
raw
history blame
No virus
617 Bytes
import pipeline
import settings
def test_need_to_update():
table_list = settings.TABLE_NAME_AND_FREQ
for table, freq in table_list:
result = pipeline.need_to_update(table, freq)
print(result)
# test_need_to_update()
def test_fetch_stock_price():
df = pipeline.fetch_stock_price()
return df
def test_need_to_update_stocks_price():
print(pipeline.need_to_update_stocks_price())
def test_add_details_to_stock_df():
stock_df = pipeline.update_stocks_price()
stock_df = pipeline.add_details_to_stock_df(stock_df)
return stock_df
print(test_add_details_to_stock_df())