Merge pull request #346 from n1rna/improve-list-tickers

Add support for multi tickers in list_tickers function
This commit is contained in:
Cameron Yick
2019-10-20 15:50:54 -04:00
committed by GitHub
2 changed files with 25 additions and 5 deletions

View File

@@ -107,6 +107,22 @@ class TestTickerPrices(TestCase):
frequency="30Min")
self.assertGreater(len(prices), 1)
@vcr.use_cassette('tests/fixtures/list_stock_tickers.yaml')
def test_list_all_stock_tickers(self):
tickers = self._client.list_tickers()
assert len(tickers) > 1
assert any(ticker['assetType' == 'Stock' for ticker in tickers)
assert any(ticker['assetType' == 'Mutual Fund' for ticker in tickers)
assert any(ticker['assetType' == 'ETF' for ticker in tickers)
@vcr.use_cassette('tests/fixtures/list_stock_tickers.yaml')
def test_list_multi_stock_tickers(self):
tickers = self._client.list_tickers(['Stock', 'ETF'])
assert len(tickers) > 1
assert any(ticker['assetType' == 'Stock' for ticker in tickers)
assert any(ticker['assetType' == 'ETF' for ticker in tickers)
assert all(ticker['assetType' != 'Mutual Fund' for ticker in tickers)
@vcr.use_cassette('tests/fixtures/list_stock_tickers.yaml')
def test_list_stock_tickers(self):
tickers = self._client.list_stock_tickers()