Update tests to match

kivy_mainline_and_py3.8
Tyler Goodlet 2018-11-13 12:57:46 -05:00
parent d145a5a219
commit 36cf68dc0f
1 changed files with 2 additions and 3 deletions

View File

@ -111,7 +111,6 @@ async def test_option_contracts(tmx_symbols):
id, contracts = await client.option_contracts(symbol) id, contracts = await client.option_contracts(symbol)
assert isinstance(id, int) assert isinstance(id, int)
assert isinstance(contracts, dict) assert isinstance(contracts, dict)
ordered = sorted(contracts)
for dt in contracts: for dt in contracts:
assert dt.isoformat( assert dt.isoformat(
timespec='microseconds') == contracts[dt]['expiryDate'] timespec='microseconds') == contracts[dt]['expiryDate']
@ -123,9 +122,9 @@ async def test_option_chain(tmx_symbols):
""" """
async with qt.get_client() as client: async with qt.get_client() as client:
# contract lookup - should be cached # contract lookup - should be cached
ids, max_expiry = await client.max_contract_expiry(tmx_symbols) contracts = await client.get_contracts(tmx_symbols)
# chains quote for all symbols # chains quote for all symbols
quotes = await client.option_chains(ids, max_expiry) quotes = await client.option_chains(contracts)
for key in tmx_symbols: for key in tmx_symbols:
contracts = quotes.pop(key) contracts = quotes.pop(key)
for key, quote in contracts.items(): for key, quote in contracts.items():