Add support for multi tickers in list_tickers function

This commit is contained in:
n1rna
2019-10-18 19:06:35 +03:30
parent 5490a3c85c
commit c3394c8cdc
2 changed files with 23 additions and 5 deletions

View File

@@ -107,6 +107,21 @@ 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)
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()