def test_dict(self):
ts_dict = get_time_series(self.keys_to_check[0])
self.assertTrue(type(ts_dict) is dict)
self.assertEqual(ts_dict.keys()[0], self.keys_to_check[0])
self.assertTrue(
type(ts_dict[self.keys_to_check[0]]) is pandas.core.series.Series)
def test_invalid_api_key(self):
with self.assertRaisesRegexp(KofdataError, 'Could not authenticate'):
get_time_series('doesntmatter', 'notakey')
def test_nonexistent_key(self):
with self.assertRaisesRegexp(KofdataError, 'The API responded'):
get_time_series('nonexistend_key')
def test_multiple_keys(self):
ts = get_time_series(self.keys_to_check)
self.assertEqual(len(ts), 2)
def test_df(self):
ts_df = get_time_series(self.keys_to_check[0], as_data_frame=True)
self.assertTrue(type(ts_df) is pandas.core.frame.DataFrame)
self.assertEqual(ts_df.columns[0], self.keys_to_check[0])