diff --git a/tests/test_tiingo.py b/tests/test_tiingo.py index 0d0da23..9d920c2 100644 --- a/tests/test_tiingo.py +++ b/tests/test_tiingo.py @@ -6,7 +6,7 @@ from unittest import TestCase import vcr from tiingo import TiingoClient -from tiingo.api import InvalidFrequencyError +from tiingo.exceptions import InvalidFrequencyError from tiingo.restclient import RestClientError diff --git a/tests/test_tiingo_pandas.py b/tests/test_tiingo_pandas.py index 54ed91e..7d22ba7 100644 --- a/tests/test_tiingo_pandas.py +++ b/tests/test_tiingo_pandas.py @@ -4,7 +4,7 @@ import vcr from unittest import TestCase from tiingo import TiingoClient -from tiingo.api import APIColumnNameError, InstallPandasException +from tiingo.exceptions import APIColumnNameError, InstallPandasException, MissingRequiredArgumentError try: import pandas as pd pandas_is_installed = True @@ -69,6 +69,10 @@ class TestTiingoWithPython(TestCase): self._client.get_dataframe(['GOOGL', 'AAPL'], startDate='2018-01-05', endDate='2018-01-19', metric_name='xopen', frequency='weekly') + def test_metric_name_missing_when_multiple_tickers(self): + with self.assertRaises(MissingRequiredArgumentError): + self._client.get_dataframe(['GOOGL', 'AAPL'], frequency='weekly') + @vcr.use_cassette('tests/fixtures/ticker_price_pandas_single.yaml') def test_pandas_edge_case(self): """Test single price/date being returned as a frame"""