Refactor News Method

This commit is contained in:
Cameron Yick
2017-09-17 15:21:03 -04:00
parent 14d3bb6bd3
commit febffd76da
2 changed files with 10 additions and 16 deletions

View File

@@ -90,16 +90,19 @@ class TestNews(TestCase):
def test_get_news_articles(self):
"""Confirm that news article work"""
NUM_ARTICLES = 10
search_params = {
"tickers": ["aapl", "googl"],
"tags": ["Technology", "Bitcoin"],
"startDate": "2016-01-01",
"endDate": "2017-08-31",
"sources": ['washingtonpost.com', 'altcointoday.com'],
"limit": 10
"limit": NUM_ARTICLES
}
articles = self._client.get_news(**search_params)
assert len(articles) == NUM_ARTICLES
for article in articles:
assert all(key in article for key in self.article_keys)

View File

@@ -109,22 +109,13 @@ class TiingoClient(RestClient):
params = {
'limit': limit,
'offset': offset,
'sortBy': sortBy
'sortBy': sortBy,
'tickers': tickers,
'sources': sources,
'tags': tags,
'startDate': startDate,
'endDate': endDate
}
# TBD: whether these commas are necessary if just pass list instead
if tickers:
params['tickers'] = ",".join(tickers)
if tags:
params['tags'] = ",".join(tags)
if sources:
params['sources'] = ",".join(sources)
if startDate:
params['startDate'] = startDate
if endDate:
params['endDate'] = endDate
response = self._request('GET', url, params=params)
return response.json()