COMPAT, TST: Remove tm.TestCase from testing (#320)

* COMPAT: Use assert_raises_regex in testing

assert_raises_regex is the renamed version
of assertRaisesRegexp from pandas < 0.20.0.

* MAINT: Remove tm.TestCase from testing

Author: gfyoung

pandas_datareader/compat/__init__.py

@@ -8,12 +8,18 @@
 PANDAS_VERSION = LooseVersion(pd.__version__)
 
 PANDAS_0190 = (PANDAS_VERSION >= LooseVersion('0.19.0'))
+PANDAS_0200 = (PANDAS_VERSION >= LooseVersion('0.20.0'))
 
 if PANDAS_0190:
     from pandas.api.types import is_number
 else:
     from pandas.core.common import is_number
 
+if PANDAS_0200:
+    from pandas.util.testing import assert_raises_regex
+else:
+    from pandas.util.testing import assertRaisesRegexp as assert_raises_regex
+
 if compat.PY3:
     from urllib.error import HTTPError
 else:

pandas_datareader/io/tests/test_jsdmx.py

@@ -9,9 +9,9 @@
 from pandas_datareader.io import read_jsdmx
 
 
-class TestJSDMX(tm.TestCase):
+class TestJSDMX(object):
 
-    def setUp(self):
+    def setup_method(self, method):
         self.dirpath = tm.get_data_path()
 
     def test_tourism(self):

pandas_datareader/io/tests/test_sdmx.py

@@ -9,9 +9,9 @@
 from pandas_datareader.io.sdmx import read_sdmx, _read_sdmx_dsd
 
 
-class TestSDMX(tm.TestCase):
+class TestSDMX(object):
 
-    def setUp(self):
+    def setup_method(self, method):
         self.dirpath = tm.get_data_path()
 
     def test_tourism(self):

pandas_datareader/tests/test_base.py

@@ -1,9 +1,8 @@
 import pytest
-import pandas.util.testing as tm
 import pandas_datareader.base as base
 
 
-class TestBaseReader(tm.TestCase):
+class TestBaseReader(object):
     def test_valid_retry_count(self):
         with pytest.raises(ValueError):
             base._BaseReader([], retry_count='stuff')
@@ -21,7 +20,7 @@ def test_invalid_format(self):
             b._read_one_data('a', None)
 
 
-class TestDailyBaseReader(tm.TestCase):
+class TestDailyBaseReader(object):
     def test_get_params(self):
         with pytest.raises(NotImplementedError):
             b = base._DailyBaseReader()

pandas_datareader/tests/test_data.py

@@ -7,21 +7,14 @@
 from pandas_datareader.data import DataReader
 
 
-class TestOptionsWarnings(tm.TestCase):
-    @classmethod
-    def setUpClass(cls):
-        super(TestOptionsWarnings, cls).setUpClass()
-
-    @classmethod
-    def tearDownClass(cls):
-        super(TestOptionsWarnings, cls).tearDownClass()
+class TestOptionsWarnings(object):
 
     def test_options_source_warning(self):
         with tm.assert_produces_warning():
             web.Options('aapl')
 
 
-class TestDataReader(tm.TestCase):
+class TestDataReader(object):
     def test_read_yahoo(self):
         gs = DataReader("GS", "yahoo")
         assert isinstance(gs, DataFrame)

pandas_datareader/tests/test_edgar.py

@@ -7,7 +7,7 @@
 from pandas_datareader._utils import RemoteDataError
 
 
-class TestEdgarIndex(tm.TestCase):
+class TestEdgarIndex(object):
 
     @classmethod
     def setup_class(cls):

pandas_datareader/tests/test_enigma.py

@@ -3,19 +3,16 @@
 
 from requests.exceptions import HTTPError
 
-import pandas.util.testing as tm
-
 import pandas_datareader as pdr
 import pandas_datareader.data as web
 
 TEST_API_KEY = os.getenv('ENIGMA_API_KEY')
 
 
-class TestEnigma(tm.TestCase):
+class TestEnigma(object):
 
     @classmethod
-    def setUpClass(cls):
-        super(TestEnigma, cls).setUpClass()
+    def setup_class(cls):
         pytest.importorskip("lxml")
 
     def test_enigma_datareader(self):

pandas_datareader/tests/test_eurostat.py

@@ -3,8 +3,10 @@
 import pandas.util.testing as tm
 import pandas_datareader.data as web
 
+from pandas_datareader.compat import assert_raises_regex
 
-class TestEurostat(tm.TestCase):
+
+class TestEurostat(object):
 
     def test_get_cdh_e_fos(self):
         # Employed doctorate holders in non managerial and non professional
@@ -87,7 +89,7 @@ def test_get_nrg_pc_202(self):
     def test_get_prc_hicp_manr_exceeds_limit(self):
         # see gh-149
         msg = 'Query size exceeds maximum limit'
-        with tm.assertRaisesRegexp(ValueError, msg):
+        with assert_raises_regex(ValueError, msg):
             web.DataReader('prc_hicp_manr', 'eurostat',
                            start=pd.Timestamp('2000-01-01'),
                            end=pd.Timestamp('2013-01-01'))

pandas_datareader/tests/test_famafrench.py

@@ -7,7 +7,7 @@
 from pandas_datareader.famafrench import get_available_datasets
 
 
-class TestFamaFrench(tm.TestCase):
+class TestFamaFrench(object):
 
     def test_get_data(self):
         keys = [

pandas_datareader/tests/test_fred.py

@@ -11,7 +11,7 @@
 from pandas_datareader._utils import RemoteDataError
 
 
-class TestFred(tm.TestCase):
+class TestFred(object):
     def test_fred(self):
 
         # Raises an exception when DataReader can't