Update price history test assertions

This commit is contained in:
Cameron Yick
2019-10-19 18:58:32 -04:00
parent 6e36317961
commit d09e21c1d7

View File

@@ -18,22 +18,23 @@ class TestCryptoEndpoints(TestCase):
def test_crypto_metadata(self): def test_crypto_metadata(self):
metadata = self._client.get_crypto_metadata(tickers=['btcusd', 'fldcbtc']) metadata = self._client.get_crypto_metadata(tickers=['btcusd', 'fldcbtc'])
assert len(metadata) == 2 assert len(metadata) == 2 # 1 item per ticker
assert metadata[0]['ticker'] == 'btcusd' assert metadata[0]['ticker'] == 'btcusd'
assert metadata[1]['ticker'] == 'fldcbtc' assert metadata[1]['ticker'] == 'fldcbtc'
@vcr.use_cassette('tests/fixtures/crypto_top_of_book.yaml') @vcr.use_cassette('tests/fixtures/crypto_top_of_book.yaml')
def test_crypto_top_of_book(self): def test_crypto_top_of_book(self):
top_of_book = self._client.get_crypto_top_of_book(tickers=['btcusd', 'fldcbtc'], includeRawExchangeData=True) top_of_book = self._client.get_crypto_top_of_book(tickers=['btcusd', 'fldcbtc'], includeRawExchangeData=True)
assert 'topOfBookData' in top_of_book assert len(top_of_book) == 2 # 1 each for btcusd and fldcbtc
assert len(top_of_book['topOfBookData']) == 2 first_ticker_data = top_of_book[0]
assert 'exchangeData' in top_of_book assert 'topOfBookData' in first_ticker_data
assert 'exchangeData' in first_ticker_data
@vcr.use_cassette('tests/fixtures/crypto_price_history.yaml') @vcr.use_cassette('tests/fixtures/crypto_price_history.yaml')
def test_crypto_price_history(self): def test_crypto_price_history(self):
price_history = self._client.get_crypto_price_history(tickers=['btcusd', 'fldcbtc'], price_history = self._client.get_crypto_price_history(tickers=['btcusd', 'fldcbtc'],
startDate='2019-05-01', endDate='2019-05-02', startDate='2019-05-01', endDate='2019-05-02',
includeRawExchangeData=True, resampleFreq='1day') includeRawExchangeData=True, resampleFreq='1day')
assert len(price_history) == 2 assert len(price_history) == 2 # 1 item per ticker
assert len(price_history[0]['priceData']) in [6, 7] assert len(price_history[0]['priceData']) == 2 # 1 set of data per day
assert 'exchangeData' in price_history[0] assert 'exchangeData' in price_history[0] # includedRawExchangeData