def test_avg_open_close_returned():
res = main(cli_args=("--securities COF GOOGL MSFT --start-date-range "
"2017-01-01 --end-date-range 2017-03-01").split())
assert len(res) == 1 # results list only contains one dict result
assert list(res[0].keys()) == ['COF', 'GOOGL', 'MSFT']
assert len(res[0]['COF']) == 3 # one for each month included
assert res[0]['COF'][0]['month'] == '2017-01'
assert len(res[0]['COF'][0]['average_open']) == 6 # ensure form is $xx.xx
def test_biggest_loser_returned():
res = main(cli_args=("--securities COF GOOGL MSFT --biggest-loser "
"--start-date-range 2017-01-01 "
"--end-date-range 2017-06-31").split())
assert len(res) == 2 # results must contain avg open/close + biggest-loser
assert len(res[1].keys()) == 1 # there can only be one biggest loser
# Because I know COF was the biggest loser in this time period, ill use it
assert list(res[1]['COF'].keys()) == ['num_days']
assert type(res[1]['COF']['num_days']) == int
def test_max_daily_profit_returned():
res = main(cli_args=("--securities COF GOOGL MSFT --max-daily-profit "
"--start-date-range 2017-01-01 "
"--end-date-range 2017-03-01").split())
assert len(res) == 2 # results must contain avg open/close + max-profit
assert sorted(list(res[1].keys())) == ['COF', 'GOOGL', 'MSFT']
for date_profit_dict in res[1]['COF']:
assert list(date_profit_dict.keys()) == ['date', 'profit']
assert len(date_profit_dict['profit']) == 5 # ensure form is $x.xx
assert len(date_profit_dict['date']) == 10 # ensure form is YYY-MM-DD
def test_busy_day_returned():
res = main(cli_args=("--securities COF GOOGL MSFT --busy-day "
"--start-date-range 2017-01-01 "
"--end-date-range 2017-03-01").split())
assert len(res) == 2 # results must contain avg open/close + busy-day
assert sorted(list(res[1].keys())) == ['COF', 'GOOGL', 'MSFT']
for k in res[1]['COF'].keys():
assert k.startswith('avg_volume_')
# there will be lots of {'date': date, 'month': month} dicts in a list.
# pull any one and make sure it follows the correct formatting
assert sorted(res[1]['COF'][k][0].keys()) == ['date', 'volume']
assert len(res[1]['COF'][k][0]['date']) == 10 # ensure form is YYYY_MM-DD
assert type(res[1]['COF'][k][0]['volume']) == float
def test_combo_returned(analyses):
num_analyses = len(analyses)
analyses = " ".join(a for a in analyses)
res = main(cli_args="--securities COF GOOGL MSFT "
"{}".format(analyses).split())
assert len(res) == num_analyses + 1 # avg monthly open/close always given
def test_invalid_securites_caught():
with pytest.raises(ValueError):
main(cli_args=("--securities CONF goog NsFT").split())
def test_incorrectly_named_args_caught():
with pytest.raises(SystemExit):
main(cli_args="--start_date_range 2017-01-01".split())
with pytest.raises(SystemExit):
main(cli_args="--security MSFT --start-date-range 2017-01-01".split())
def test_invalid_dates_caught():
with pytest.raises(ValueError):
main(cli_args=("--start-date-range 1-1-17").split())
with pytest.raises(ValueError):
main(cli_args=("--end-date-range 6/31/17").split())