def test_between_call_connection_with_datetimes(self, mock):
quandl.get_point_in_time(
'ZACKS/FC',
interval='between',
start_date='2020-01-01T12:00',
end_date='2020-01-02T14:00'
)
expected = call('get', 'pit/ZACKS/FC/between/2020-01-01T12:00/2020-01-02T14:00', params={})
self.assertEqual(mock.call_args, expected)
def test_from_call_connection(self, mock):
quandl.get_point_in_time(
'ZACKS/FC',
interval='from',
start_date='2020-01-01',
end_date='2020-01-02'
)
expected = call('get', 'pit/ZACKS/FC/from/2020-01-01/to/2020-01-02', params={})
self.assertEqual(mock.call_args, expected)
def test_invalid_between_connection(self, mock):
self.assertRaises(
InvalidRequestError,
lambda: quandl.get_point_in_time('ZACKS/FC', interval='between'))
self.assertRaises(
InvalidRequestError, lambda: quandl.get_point_in_time(
'ZACKS/FC', interval='between', start_date='2020-01-01'))
self.assertRaises(
InvalidRequestError, lambda: quandl.get_point_in_time(
'ZACKS/FC', interval='between', end_date='2020-01-02'))
def test_invalid_interval_connection(self, mock):
self.assertRaises(InvalidRequestError, lambda: quandl.get_point_in_time('ZACKS/FC'))
self.assertRaises(
InvalidRequestError,
lambda: quandl.get_point_in_time('ZACKS/FC', interval='quandl')
)
def test_asofdate_call_without_date(self, mock):
quandl.get_point_in_time('ZACKS/FC', interval='asofdate')
expected = call('get', "pit/ZACKS/FC/asofdate/%s" % date.today(), params={})
self.assertEqual(mock.call_args, expected)
def test_asofdate_call_connection_with_datetimes(self, mock):
quandl.get_point_in_time('ZACKS/FC', interval='asofdate', date='2020-01-01T12:55')
expected = call('get', 'pit/ZACKS/FC/asofdate/2020-01-01T12:55', params={})
self.assertEqual(mock.call_args, expected)
def test_get_point_in_time_returns_data_frame_object(self, mock):
df = quandl.get_point_in_time('ZACKS/FC', interval='asofdate', date='2020-01-01')
self.assertIsInstance(df, pandas.core.frame.DataFrame)
