aljaz/tiingo-python
1
1
Fork 0
mirror of https://github.com/hydrosquall/tiingo-python.git synced 2025-12-21 05:14:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix tests

Browse Source
This commit is contained in:
Koen werklaptop
2024-11-03 20:01:54 +01:00
parent 027bd0d1c6
commit 27e634d807
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_tiingo.py
View File

@@ -115,10 +115,12 @@ class TestTickerPrices(TestCase):
@vcr.use_cassette('tests/fixtures/ticker_price_with_multiple_columns.yaml')
def test_ticker_price_with_multiple_columns(self):
"""Confirm that requesting specific columns works"""
requested_columns = "open,high,low,close,volume"
prices = self._client.get_ticker_price("GOOGL",
columns="open,high,low,close,volume",
columns=requested_columns,
fmt='json')
assert len(prices) == 1
assert len(prices[0]) == len(requested_columns.split(',')) + 1
assert prices[0].get('date')
assert prices[0].get('high')
assert prices[0].get('low')

2
tests/test_tiingo_pandas.py
View File

@@ -132,7 +132,7 @@ class TestTiingoWithPython(TestCase):
columns=requested_columns,
fmt='json')
assert len(prices) == 1
assert len(prices.columns) == len(requested_columns.split(",")) + 1
assert len(prices.columns) == len(requested_columns.split(','))
def test_metric_name_column_error(self):
with self.assertRaises(APIColumnNameError):