def test_error_subscription(self):
contents = {'quandl_error': {
'code': 'QEPx04',
'message': ('You do not have permission to view this dataset. '
'Please subscribe to this database to get '
'access.')}}
with response(contents):
with self.assertRaises(ValueError) as exc:
quandl.fetch_time_series('WIKI:FB', None)
self.assertRegex(exc.message, 'premium')
def test_error_premium(self):
contents = {'quandl_error': {
'code': 'QEPx05',
'message': ('You have attempted to view a premium database in '
'anonymous mode, i.e., without providing a Quandl '
'key. Please register for a free Quandl account, '
'and then include your API key with your '
'requests.')}}
with response(contents):
with self.assertRaises(ValueError) as exc:
quandl.fetch_time_series('WIKI:FB', None)
self.assertRegex(exc.message, 'premium')
def test_non_standard_columns(self):
contents = {
'dataset': {'collapse': None,
'column_index': None,
'column_names': ['Date',
'USD (AM)',
'USD (PM)',
'GBP (AM)',
'GBP (PM)',
'EURO (AM)',
'EURO (PM)'],
'data': [['2019-06-18',
1344.55,
1341.35,
1073.22,
1070.67,
1201.89,
1198.09]],
'end_date': '2019-06-18',
'frequency': 'daily',
'order': None,
'limit': 1,
'start_date': '2019-06-08',
'transform': None}}
with response(contents):
srcprice = quandl.fetch_time_series('LBMA:GOLD:USD_(PM)', None)
self.assertIsInstance(srcprice, source.SourcePrice)
self.assertEqual(D('1341.35'), srcprice.price)
self.assertEqual(datetime.datetime(2019, 6, 18, 0, 0, 0,
tzinfo=tz.tzutc()),
srcprice.time.astimezone(tz.tzutc()))
self.assertEqual(None, srcprice.quote_currency)
def _test_valid_response(self):
contents = {
'dataset': {'collapse': None,
'column_index': None,
'column_names': ['Date',
'Open',
'High',
'Low',
'Close',
'Volume',
'Ex-Dividend',
'Split Ratio',
'Adj. Open',
'Adj. High',
'Adj. Low',
'Adj. Close',
'Adj. Volume'],
'data': [['2018-03-27',
1063.9,
1064.54,
997.62,
1006.94,
2940957.0,
0.0,
1.0,
1063.9,
1064.54,
997.62,
1006.94,
2940957.0]],
'database_code': 'WIKI',
'database_id': 4922,
'dataset_code': 'GOOGL',
'description': 'This dataset has no description.',
'end_date': '2018-03-27',
'frequency': 'daily',
'id': 11304017,
'limit': 1,
'name': 'Alphabet Inc (GOOGL) Prices, Dividends, Splits and '
'Trading Volume',
'newest_available_date': '2018-03-27',
'oldest_available_date': '2004-08-19',
'order': None,
'premium': False,
'refreshed_at': '2018-03-27T21:46:11.201Z',
'start_date': '2004-08-19',
'transform': None,
'type': 'Time Series'}}
with response(contents):
srcprice = quandl.fetch_time_series('WIKI:FB', None)
self.assertIsInstance(srcprice, source.SourcePrice)
self.assertEqual(D('1006.94'), srcprice.price)
self.assertEqual(datetime.datetime(2018, 3, 27, 0, 0, 0,
tzinfo=tz.tzutc()),
srcprice.time.astimezone(tz.tzutc()))
self.assertEqual(None, srcprice.quote_currency)
def test_error_network(self):
with response(None, 404):
with self.assertRaises(ValueError) as exc:
quandl.fetch_time_series('WIKI:FB', None)
self.assertRegex(exc.message, 'premium')