def test_invalid_symbol_or_symbols(self):
with pytest.raises(IEXSymbolError):
a = Stock(["TSLA", "BAD SYMBOL", "BAD SYMBOL"])
a.get_price()
def test_invalid_symbol(self):
data = Stock("BAD SYMBOL")
with pytest.raises(IEXSymbolError):
data.get_price()
class TestBatchDefault(object):
def setup_class(self):
self.cbatch = Stock(["aapl", "tsla"])
self.cbatch2 = Stock(["aapl", "tsla"], output_format='pandas')
self.cbatch3 = Stock(["uvxy", "svxy"])
def test_invalid_symbol_or_symbols(self):
with pytest.raises(IEXSymbolError):
a = Stock(["TSLA", "BAD SYMBOL", "BAD SYMBOL"])
a.get_price()
def test_get_endpoints(self):
data = self.cbatch.get_endpoints(["price"])["AAPL"]
assert list(data) == ["price"]
def test_get_endpoints_bad_endpoint(self):
with pytest.raises(IEXEndpointError):
self.cbatch.get_endpoints(["BAD ENDPOINT", "quote"])
with pytest.raises(IEXEndpointError):
self.cbatch.get_endpoints("BAD ENDPOINT")
def test_get_all(self):
data = self.cbatch.get_all()
assert len(data) == 2
assert len(data["AAPL"]) == 20
def test_get_all_format(self):
data = self.cbatch.get_all()
assert isinstance(data, dict)
data2 = self.cbatch2.get_all()
assert isinstance(data2["AAPL"], dict)
def test_get_book_format(self):
data = self.cbatch.get_book()
assert isinstance(data, dict)
data2 = self.cbatch2.get_book()
assert isinstance(data2, pd.DataFrame)
def test_get_chart_format(self):
data = self.cbatch.get_chart()
assert isinstance(data, dict)
data2 = self.cbatch2.get_chart()
assert isinstance(data2["AAPL"], list)
def test_get_chart_params(self):
data = self.cbatch.get_chart()["AAPL"]
# Test chart range_s
data2 = self.cbatch.get_chart(range_='1y')["AAPL"]
assert 15 < len(data) < 35
assert 240 < len(data2) < 260
# Test chartSimplify
data4 = self.cbatch.get_chart(chartSimplify=True)["AAPL"][0]
assert "simplifyFactor" in list(data4)
data5 = self.cbatch.get_chart(range_='1y', chartInterval=5)["AAPL"]
assert 45 < len(data5) < 55
@pytest.mark.xfail(reason="This test only works overnight")
def test_get_chart_reset(self):
# Test chartReset
data = self.cbatch.get_chart(range_='1d', chartReset=True)
assert data == []
def test_get_company_format(self):
data = self.cbatch.get_company()
assert isinstance(data, dict)
data2 = self.cbatch2.get_company()
assert isinstance(data2, pd.DataFrame)
def test_get_delayed_quote_format(self):
data = self.cbatch.get_delayed_quote()
assert isinstance(data, dict)
data2 = self.cbatch2.get_delayed_quote()
assert isinstance(data2, pd.DataFrame)
def test_get_dividends_format(self):
data = self.cbatch.get_dividends()
assert isinstance(data, dict)
data2 = self.cbatch2.get_dividends()
assert isinstance(data2, dict)
def test_get_dividends_params(self):
data = self.cbatch.get_dividends()["AAPL"]
data2 = self.cbatch.get_dividends(range_='2y')["AAPL"]
data3 = self.cbatch.get_dividends(range_='5y')["AAPL"]
assert len(data) < len(data2) < len(data3)
def test_get_earnings_format(self):
data = self.cbatch.get_earnings()
assert isinstance(data, dict)
data2 = self.cbatch2.get_earnings()
assert isinstance(data2, pd.DataFrame)
def test_get_effective_spread_format(self):
data = self.cbatch.get_effective_spread()
assert isinstance(data, dict)
data2 = self.cbatch2.get_effective_spread()
assert isinstance(data2, pd.DataFrame)
def test_get_financials_format(self):
data = self.cbatch.get_financials()
assert isinstance(data, dict)
data2 = self.cbatch2.get_financials()
assert isinstance(data2, pd.DataFrame)
def test_get_key_stats_format(self):
data = self.cbatch.get_key_stats()
assert isinstance(data, dict)
data2 = self.cbatch2.get_key_stats()
assert isinstance(data2, pd.DataFrame)
def test_get_logo_format(self):
data = self.cbatch.get_logo()
assert isinstance(data, dict)
data2 = self.cbatch2.get_logo()
assert isinstance(data2, pd.DataFrame)
def test_get_news_format(self):
data = self.cbatch.get_news()
assert isinstance(data, dict)
def test_ohlc(self):
data = self.cbatch.get_ohlc()
assert isinstance(data, dict)
def test_get_open_close_format(self):
data = self.cbatch.get_open_close()
assert isinstance(data, dict)
data2 = self.cbatch2.get_open_close()
assert isinstance(data2, pd.DataFrame)
def test_get_peers_format(self):
data = self.cbatch.get_peers()
assert isinstance(data, dict)
data2 = self.cbatch2.get_peers()
assert isinstance(data2, dict)
def test_get_previous_format(self):
data = self.cbatch.get_previous()
assert isinstance(data, dict)
data2 = self.cbatch2.get_previous()
assert isinstance(data2, pd.DataFrame)
def test_get_price_format(self):
data = self.cbatch.get_price()
assert isinstance(data, dict)
data2 = self.cbatch2.get_price()
assert isinstance(data2, pd.DataFrame)
def test_get_quote_format(self):
data = self.cbatch.get_quote()
assert isinstance(data, dict)
data2 = self.cbatch2.get_quote()
assert isinstance(data2, pd.DataFrame)
data3 = self.cbatch.get_quote(displayPercent=True)
assert (abs(data3["AAPL"]["ytdChange"]) > abs(
data["AAPL"]["ytdChange"]))
def test_get_relevant_format(self):
data = self.cbatch.get_relevant()
assert isinstance(data, dict)
data2 = self.cbatch2.get_relevant()
assert isinstance(data2, pd.DataFrame)
def test_get_splits(self):
data = self.cbatch.get_splits()
assert isinstance(data, dict)
data2 = self.cbatch2.get_splits()
assert isinstance(data2, pd.DataFrame)
def test_get_splits_params(self):
data = self.cbatch3.get_splits(range_="2y")["SVXY"]
data2 = self.cbatch3.get_splits(range_="5y")["SVXY"]
assert len(data2) > len(data)
def test_time_series(self):
data = self.cbatch.get_time_series()
data2 = self.cbatch.get_chart()
assert data == data2
def test_get_volume_by_venue_format(self):
data = self.cbatch.get_volume_by_venue()
assert isinstance(data, dict)
data2 = self.cbatch2.get_volume_by_venue()
assert isinstance(data2, pd.DataFrame)
def test_get_select_ep_bad_params(self):
with pytest.raises(ValueError):
self.cbatch.get_endpoints()
with pytest.raises(IEXEndpointError):
self.cbatch.get_endpoints("BADENDPOINT")