def test_add_remove_stocks():
tickers = [
random_utilities.random_letters(),
random_utilities.random_letters()
]
assert tickers_table.add_tickers(tickers) is True
assert tickers_table.remove_tickers(tickers) is True
def test_add_tickers_one_duplicate():
new_ticker = random_utilities.random_letters()
tickers_table.add_ticker('test')
tickers = ['test', new_ticker]
assert tickers_table.add_tickers(tickers) is True
retrieved_tickers = tickers_table.get_tickers()
assert new_ticker in retrieved_tickers
def test_remove_row(cursor):
column = 'id'
columns = f'({column})'
value = random_utilities.random_letters()
values = f"('{value}')"
postgres.insert_row(cursor, test_table_name, columns, values)
assert postgres.remove_row(cursor, test_table_name, column, value) is True
def add_random_row():
row = {
'ticker': random_utilities.random_letters(),
'company_profile': random_utilities.random_date(),
'daily_history': random_utilities.random_date()
}
api_progress_table.insert_row(row)
def test_get_tickers():
ticker = random_utilities.random_letters()
tickers_table.add_ticker(ticker)
tickers = tickers_table.get_tickers()
assert tickers is not None
assert len(tickers) > 0
assert ticker in tickers
for ticker in tickers:
assert isinstance(ticker, str)
assert len(ticker) > 0
def _random_row(ticker=None):
if ticker is None:
ticker = random_utilities.random_letters()
return {
'ticker': ticker,
'date': random_utilities.random_date(),
'open': random_utilities.random_double(),
'high': random_utilities.random_double(),
'low': random_utilities.random_double(),
'close': random_utilities.random_double(),
'adjusted_close': random_utilities.random_double(),
'volume': random_utilities.random_int(0, 10000000),
'dividend': random_utilities.random_double(),
'split_coefficient': 1.000
}
def test_website():
website = random_utilities.random_letters(8)
assert company_profile_table.update_website(test_ticker, website)
retrieved_website = company_profile_table.get_website(test_ticker)
assert retrieved_website == website
def test_update_row_string(cursor):
new_value = random_utilities.random_letters()
assert postgres.update_value(cursor, test_table_name, 'id', test_table_row_id, 'id', new_value) is True
def test_country():
country = random_utilities.random_letters(2).upper()
assert company_profile_table.update_country(test_ticker, country)
retrieved_country = company_profile_table.get_country(test_ticker)
assert retrieved_country == country
def test_delete_column(cursor):
column_name = random_utilities.random_letters()
postgres.add_column(cursor, test_table_name, column_name)
assert postgres.delete_column(cursor, test_table_name, column_name) is True
def test_insert_row_dict_invalid(cursor):
values = {'invalid': random_utilities.random_letters(10)}
assert postgres.insert_row_as_dict(cursor, test_table_name, values) is False
def test_random_letters_valid():
random_letters = random_utilities.random_letters()
for letter in random_letters:
assert letter in string.ascii_letters
def test_name():
name = random_utilities.random_letters(12)
assert company_profile_table.update_name(test_ticker, name)
retrieved_name = company_profile_table.get_name(test_ticker)
assert retrieved_name == name
def test_add_tickers():
tickers = []
for _ in range(1, 3):
tickers.append(random_utilities.random_letters())
assert api_progress_table.add_tickers(tickers) is True
def test_add_stock():
ticker = random_utilities.random_letters(12)
assert api_progress_table.add_ticker(ticker) is True
def test_description():
description = random_utilities.random_letters(8)
assert company_profile_table.update_description(test_ticker, description)
retrieved_description = company_profile_table.get_description(test_ticker)
assert retrieved_description == description
def test_ceo():
ceo = random_utilities.random_letters(8)
assert company_profile_table.update_ceo(test_ticker, ceo)
retrieved_ceo = company_profile_table.get_ceo(test_ticker)
assert retrieved_ceo == ceo
def test_industry():
industry = random_utilities.random_letters(8)
assert company_profile_table.update_industry(test_ticker, industry)
retrieved_industry = company_profile_table.get_industry(test_ticker)
assert retrieved_industry == industry
def test_sector():
sector = random_utilities.random_letters(8)
assert company_profile_table.update_sector(test_ticker, sector)
retrieved_sector = company_profile_table.get_sector(test_ticker)
assert retrieved_sector == sector
def test_exchange():
exchange = random_utilities.random_letters(8).upper()
assert company_profile_table.update_exchange(test_ticker, exchange)
retrieved_exchange = company_profile_table.get_exchange(test_ticker)
assert retrieved_exchange == exchange
def test_add_remove_ticker():
ticker = random_utilities.random_letters()
assert tickers_table.add_ticker(ticker) is True
assert tickers_table.remove_ticker(ticker) is True
def test_random_letters():
assert random_utilities.random_letters() is not None
def test_random_letters_length():
length = 1
random_letters = random_utilities.random_letters(length)
assert len(random_letters) == length
def test_create_table_one_column(cursor):
table_name = random_utilities.random_letters()
columns = {'id': 'varchar(10) PRIMARY KEY'}
assert postgres.create_table(cursor, table_name, columns) is True
postgres.delete_table(cursor, table_name)
def test_database_doesnt_exist(cursor):
database_name = random_utilities.random_letters()
assert postgres.database_exists(cursor, database_name) is False
def test_add_stock():
ticker = random_utilities.random_letters(12)
assert company_profile_table.add_stock(ticker)
def test_add_tickers_one_duplicate():
new_ticker = random_utilities.random_letters()
api_progress_table.add_ticker('test')
tickers = ['test', new_ticker]
assert api_progress_table.add_tickers(tickers) is True
def test_create_database():
connection = _get_connection_to_server()
cursor = connection.cursor()
name = random_utilities.random_letters()
assert postgres.create_database(cursor, name) is True
postgres.delete_database(cursor, name)
def test_update_invalid_ticker():
ticker = random_utilities.random_letters()
assert company_profile_updater.update(ticker) is False
def test_row_exists_doesnt_exist():
ticker = random_utilities.random_letters()
year = random_utilities.random_date().year
assert yearly_history_table.row_exists(ticker, year) is False
