aljaz/tiingo-python
1
1
Fork 0
mirror of https://github.com/hydrosquall/tiingo-python.git synced 2026-01-04 11:54:50 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add columns param to get_ticker_price, including tests

Browse Source
This commit is contained in:
Koen werklaptop
2024-10-23 21:27:02 +02:00
parent 80b40c5be5
commit 6b53725dca
5 changed files with 109 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
tests/test_tiingo.py
View File

@@ -55,7 +55,7 @@ class TestTickerPrices(TestCase):
def test_ticker_metadata_as_object(self):
metadata = self._client.get_ticker_metadata("GOOGL", fmt="object")
assert metadata.ticker == "GOOGL" # Access property via ATTRIBUTE
assert metadata.name # (contrast with key access above
assert metadata.name # (contrast with key access above
@vcr.use_cassette('tests/fixtures/ticker_price.yaml')
def test_ticker_price(self):
@@ -68,7 +68,7 @@ class TestTickerPrices(TestCase):
def test_ticker_price(self):
"""Test that weekly frequency works"""
prices = self._client.get_ticker_price("GOOGL", startDate='2018-01-05',
endDate='2018-01-19', frequency='weekly')
endDate='2018-01-19', frequency='weekly')
assert len(prices) == 3
assert prices[0].get('adjClose')
@@ -98,6 +98,34 @@ class TestTickerPrices(TestCase):
rows = list(reader)
assert len(rows) > 2 # more than 1 day of data
@vcr.use_cassette('tests/fixtures/ticker_price_with_volume_column.yaml')
def test_ticker_price_with_volume_column(self):
"""Confirm that CSV endpoint works"""
prices = self._client.get_ticker_price("GOOGL",
columns="volume",
fmt='json')
assert len(prices) == 1
assert prices[0].get('date')
assert not prices[0].get('high')
assert not prices[0].get('low')
assert not prices[0].get('open')
assert not prices[0].get('close')
assert prices[0].get('volume')
@vcr.use_cassette('tests/fixtures/ticker_price_with_multiple_columns.yaml')
def test_ticker_price_with_multiple_columns(self):
"""Confirm that CSV endpoint works"""
prices = self._client.get_ticker_price("GOOGL",
columns="open,high,low,close,volume",
fmt='json')
assert len(prices) == 1
assert prices[0].get('date')
assert prices[0].get('high')
assert prices[0].get('low')
assert prices[0].get('open')
assert prices[0].get('close')
assert prices[0].get('volume')
@vcr.use_cassette('tests/fixtures/intraday_price.yaml')
def test_intraday_ticker_price(self):
"""Test the EOD Prices Endpoint with data param"""
@@ -149,6 +177,7 @@ class TestTickerPrices(TestCase):
endDate="2018-01-02",
frequency="1.5mins")
# tiingo/news
class TestNews(TestCase):
@@ -227,6 +256,7 @@ class TestNews(TestCase):
with self.assertRaises(RestClientError):
assert self._client.get_bulk_news(file_id="1", fmt="object")
# FUNDAMENTALS ENDPOINTS
class TestFundamentals(TestCase):