TST: changed refs to use testing/TestCase where setUpClass/tearDownClass are defined

Author: jreback

pandas/computation/tests/test_eval.py

@@ -1,6 +1,5 @@
 #!/usr/bin/env python
 
-import unittest
 import functools
 from itertools import product
 
@@ -104,10 +103,11 @@ def _is_py3_complex_incompat(result, expected):
 _good_arith_ops = com.difference(_arith_ops_syms, _special_case_arith_ops_syms)
 
 
-class TestEvalNumexprPandas(unittest.TestCase):
+class TestEvalNumexprPandas(tm.TestCase):
 
     @classmethod
     def setUpClass(cls):
+        super(TestEvalNumexprPandas, cls).setUpClass()
         skip_if_no_ne()
         import numexpr as ne
         cls.ne = ne
@@ -116,6 +116,7 @@ def setUpClass(cls):
 
     @classmethod
     def tearDownClass(cls):
+        super(TestEvalNumexprPandas, cls).tearDownClass()
         del cls.engine, cls.parser
         if hasattr(cls, 'ne'):
             del cls.ne
@@ -707,6 +708,7 @@ class TestEvalNumexprPython(TestEvalNumexprPandas):
 
     @classmethod
     def setUpClass(cls):
+        super(TestEvalNumexprPython, cls).setUpClass()
         skip_if_no_ne()
         import numexpr as ne
         cls.ne = ne
@@ -733,6 +735,7 @@ class TestEvalPythonPython(TestEvalNumexprPython):
 
     @classmethod
     def setUpClass(cls):
+        super(TestEvalPythonPython, cls).setUpClass()
         cls.engine = 'python'
         cls.parser = 'python'
 
@@ -761,6 +764,7 @@ class TestEvalPythonPandas(TestEvalPythonPython):
 
     @classmethod
     def setUpClass(cls):
+        super(TestEvalPythonPandas, cls).setUpClass()
         cls.engine = 'python'
         cls.parser = 'pandas'
 
@@ -1024,17 +1028,19 @@ def test_performance_warning_for_poor_alignment(self):
 #------------------------------------
 # slightly more complex ops
 
-class TestOperationsNumExprPandas(unittest.TestCase):
+class TestOperationsNumExprPandas(tm.TestCase):
 
     @classmethod
     def setUpClass(cls):
+        super(TestOperationsNumExprPandas, cls).setUpClass()
         skip_if_no_ne()
         cls.engine = 'numexpr'
         cls.parser = 'pandas'
         cls.arith_ops = expr._arith_ops_syms + expr._cmp_ops_syms
 
     @classmethod
     def tearDownClass(cls):
+        super(TestOperationsNumExprPandas, cls).tearDownClass()
         del cls.engine, cls.parser
 
     def eval(self, *args, **kwargs):
@@ -1337,6 +1343,7 @@ class TestOperationsNumExprPython(TestOperationsNumExprPandas):
 
     @classmethod
     def setUpClass(cls):
+        super(TestOperationsNumExprPython, cls).setUpClass()
         if not _USE_NUMEXPR:
             raise nose.SkipTest("numexpr engine not installed")
         cls.engine = 'numexpr'
@@ -1404,6 +1411,7 @@ class TestOperationsPythonPython(TestOperationsNumExprPython):
 
     @classmethod
     def setUpClass(cls):
+        super(TestOperationsPythonPython, cls).setUpClass()
         cls.engine = cls.parser = 'python'
         cls.arith_ops = expr._arith_ops_syms + expr._cmp_ops_syms
         cls.arith_ops = filter(lambda x: x not in ('in', 'not in'),
@@ -1414,6 +1422,7 @@ class TestOperationsPythonPandas(TestOperationsNumExprPandas):
 
     @classmethod
     def setUpClass(cls):
+        super(TestOperationsPythonPandas, cls).setUpClass()
         cls.engine = 'python'
         cls.parser = 'pandas'
         cls.arith_ops = expr._arith_ops_syms + expr._cmp_ops_syms

pandas/io/tests/test_clipboard.py

@@ -1,5 +1,3 @@
-import unittest
-
 import numpy as np
 from numpy.random import randint
 
@@ -18,9 +16,10 @@
     raise nose.SkipTest("no clipboard found")
 
 
-class TestClipboard(unittest.TestCase):
+class TestClipboard(tm.TestCase):
     @classmethod
     def setUpClass(cls):
+        super(TestClipboard, cls).setupClass()
         cls.data = {}
         cls.data['string'] = mkdf(5, 3, c_idx_type='s', r_idx_type='i',
                                   c_idx_names=[None], r_idx_names=[None])
@@ -43,6 +42,7 @@ def setUpClass(cls):
 
     @classmethod
     def tearDownClass(cls):
+        super(TestClipboard, cls).tearDownClass()
         del cls.data_types, cls.data
 
     def check_round_trip_frame(self, data_type, excel=None, sep=None):

pandas/io/tests/test_data.py

@@ -1,6 +1,5 @@
 from __future__ import print_function
 from pandas import compat
-import unittest
 import warnings
 import nose
 from nose.tools import assert_equal
@@ -35,15 +34,17 @@ def assert_n_failed_equals_n_null_columns(wngs, obj, cls=SymbolWarning):
     assert msgs.str.contains('|'.join(failed_symbols)).all()
 
 
-class TestGoogle(unittest.TestCase):
+class TestGoogle(tm.TestCase):
     @classmethod
     def setUpClass(cls):
+        super(TestGoogle, cls).setupClass()
         cls.locales = tm.get_locales(prefix='en_US')
         if not cls.locales:
             raise nose.SkipTest("US English locale not available for testing")
 
     @classmethod
     def tearDownClass(cls):
+        super(TestGoogle, cls).tearDownClass()
         del cls.locales
 
     @network
@@ -105,9 +106,10 @@ def test_get_multi2(self):
                 assert_n_failed_equals_n_null_columns(w, result)
 
 
-class TestYahoo(unittest.TestCase):
+class TestYahoo(tm.TestCase):
     @classmethod
     def setUpClass(cls):
+        super(TestYahoo, cls).setupClass()
         _skip_if_no_lxml()
 
     @network
@@ -224,9 +226,10 @@ def test_get_date_ret_index(self):
         assert np.issubdtype(pan.values.dtype, np.floating)
 
 
-class TestYahooOptions(unittest.TestCase):
+class TestYahooOptions(tm.TestCase):
     @classmethod
     def setUpClass(cls):
+        super(TestYahooOptions, cls).setupClass()
         _skip_if_no_lxml()
 
         # aapl has monthlies
@@ -241,6 +244,7 @@ def setUpClass(cls):
 
     @classmethod
     def tearDownClass(cls):
+        super(TestYahooOptions, cls).tearDownClass()
         del cls.aapl, cls.expiry
 
     @network
@@ -283,9 +287,10 @@ def test_get_put_data(self):
             assert len(puts)>1
 
 
-class TestOptionsWarnings(unittest.TestCase):
+class TestOptionsWarnings(tm.TestCase):
     @classmethod
     def setUpClass(cls):
+        super(TestOptionsWarnings, cls).setupClass()
         _skip_if_no_lxml()
 
         with assert_produces_warning(FutureWarning):
@@ -300,6 +305,7 @@ def setUpClass(cls):
 
     @classmethod
     def tearDownClass(cls):
+        super(TestOptionsWarnings, cls).tearDownClass()
         del cls.aapl, cls.year, cls.month
 
     @network
@@ -342,7 +348,7 @@ def test_get_put_data_warning(self):
                 warnings.warn("IndexError thrown no tables found")
 
 
-class TestDataReader(unittest.TestCase):
+class TestDataReader(tm.TestCase):
     def test_is_s3_url(self):
         from pandas.io.common import _is_s3_url
         self.assert_(_is_s3_url("s3://pandas/somethingelse.com"))
@@ -372,7 +378,7 @@ def test_read_famafrench(self):
             assert isinstance(ff, dict)
 
 
-class TestFred(unittest.TestCase):
+class TestFred(tm.TestCase):
     @network
     def test_fred(self):
         """

pandas/io/tests/test_gbq.py

@@ -3,7 +3,6 @@
 import os
 import shutil
 import subprocess
-import unittest
 
 import numpy as np
 
@@ -41,18 +40,18 @@ def GetTableSchema(self,table_dict):
 class FakeApiClient:
     def __init__(self):
         self._fakejobs = FakeJobs()
-        
+
 
     def jobs(self):
         return self._fakejobs
 
 class FakeJobs:
-    def __init__(self): 
+    def __init__(self):
         self._fakequeryresults = FakeResults()
 
     def getQueryResults(self, job_id=None, project_id=None,
                         max_results=None, timeout_ms=None, **kwargs):
-        return self._fakequeryresults   
+        return self._fakequeryresults
 
 class FakeResults:
     def execute(self):
@@ -74,7 +73,7 @@ def execute(self):
 
 ####################################################################################
 
-class test_gbq(unittest.TestCase):
+class TestGbq(tm.TestCase):
     def setUp(self):
         with open(self.fake_job_path, 'r') as fin:
             self.fake_job = ast.literal_eval(fin.read())
@@ -102,7 +101,7 @@ def setUp(self):
              ('othello', 1603, 'brawl', 2),
              ('othello', 1603, "'", 17),
              ('othello', 1603, 'troubled', 1)
-            ], 
+            ],
             dtype=[('corpus', 'S16'),
                    ('corpus_date', '<i8'),
                    ('word', 'S16'),
@@ -137,17 +136,18 @@ def setUp(self):
           'TRUE_BOOLEAN',
           'FALSE_BOOLEAN',
           'NULL_BOOLEAN']]
-    
+
     @classmethod
     def setUpClass(self):
         # Integration tests require a valid bigquery token
         # be present in the user's home directory. This
         # can be generated with 'bq init' in the command line
+        super(TestGbq, cls).setupClass()
         self.dirpath = tm.get_data_path()
         home = os.path.expanduser("~")
         self.bq_token = os.path.join(home, '.bigquery.v2.token')
         self.fake_job_path = os.path.join(self.dirpath, 'gbq_fake_job.txt')
-        
+
         # If we're using a valid token, make a test dataset
         # Note, dataset functionality is beyond the scope
         # of the module under test, so we rely on the command
@@ -167,7 +167,7 @@ def test_valid_authentication(self):
         # If the user has a token file, they should recieve a client from gbq._authenticate
         if not os.path.exists(self.bq_token):
             raise nose.SkipTest('Skipped because authentication information is not available.')
-        
+
         self.assertTrue(gbq._authenticate is not None, 'Authentication To GBQ Failed')
 
     @with_connectivity_check
@@ -205,14 +205,14 @@ def test_data_small(self):
                         'An element in the result DataFrame didn\'t match the sample set')
 
     def test_index_column(self):
-        # A user should be able to specify an index column for return     
+        # A user should be able to specify an index column for return
         result_frame = gbq._parse_data(FakeClient(), self.fake_job, index_col='word')
         correct_frame = DataFrame(self.correct_data_small)
         correct_frame.set_index('word', inplace=True)
         self.assertTrue(result_frame.index.name == correct_frame.index.name)
 
     def test_column_order(self):
-        # A User should be able to specify the order in which columns are returned in the dataframe    
+        # A User should be able to specify the order in which columns are returned in the dataframe
         col_order = ['corpus_date', 'word_count', 'corpus', 'word']
         result_frame = gbq._parse_data(FakeClient(), self.fake_job, col_order=col_order)
         tm.assert_index_equal(result_frame.columns, DataFrame(self.correct_data_small)[col_order].columns)
@@ -279,8 +279,8 @@ def test_download_all_data_types(self):
     @with_connectivity_check
     def test_table_exists(self):
         # Given a table name in the format {dataset}.{tablename}, if a table exists,
-        # the GetTableReference should accurately indicate this. 
-        # This could possibly change in future implementations of bq, 
+        # the GetTableReference should accurately indicate this.
+        # This could possibly change in future implementations of bq,
         # but it is the simplest way to provide users with appropriate
         # error messages regarding schemas.
         if not os.path.exists(self.bq_token):
@@ -309,7 +309,7 @@ def test_upload_new_table_schema_error(self):
         df = DataFrame(self.correct_data_small)
         with self.assertRaises(gbq.SchemaMissing):
             gbq.to_gbq(df, 'pandas_testing_dataset.test_database', schema=None, col_order=None, if_exists='fail')
- 
+
     @with_connectivity_check
     def test_upload_replace_schema_error(self):
         # Attempting to replace an existing table without specifying a schema should fail
@@ -319,7 +319,7 @@ def test_upload_replace_schema_error(self):
         df = DataFrame(self.correct_data_small)
         with self.assertRaises(gbq.SchemaMissing):
             gbq.to_gbq(df, 'pandas_testing_dataset.test_database', schema=None, col_order=None, if_exists='replace')
-    
+
     @with_connectivity_check
     def test_upload_public_data_error(self):
         # Attempting to upload to a public, read-only, dataset should fail
@@ -432,7 +432,7 @@ def test_upload_replace(self):
                                        'contributor_ip','contributor_id','contributor_username','timestamp',
                                        'is_minor','is_bot','reversion_id','comment','num_characters'])
         gbq.to_gbq(df1, 'pandas_testing_dataset.test_data5', schema=schema, col_order=None, if_exists='fail')
-        
+
         array2 = [['TESTING_GBQ', 999999999, 'hi', 0, True, 9999999999, '00.000.00.000', 1, 'hola',
                  99999999, False, False, 1, 'Jedi', 11210]]
 
@@ -441,7 +441,7 @@ def test_upload_replace(self):
                                        'contributor_ip','contributor_id','contributor_username','timestamp',
                                        'is_minor','is_bot','reversion_id','comment','num_characters'])
         gbq.to_gbq(df2, 'pandas_testing_dataset.test_data5', schema=schema, col_order=None, if_exists='replace')
-        
+
         # Read the table and confirm the new data is all that is there
         a = gbq.read_gbq("SELECT * FROM pandas_testing_dataset.test_data5")
         self.assertTrue((a == df2).all().all())

pandas/io/tests/test_html.py

@@ -3,7 +3,6 @@
 import os
 import re
 import warnings
-import unittest
 
 try:
     from importlib import import_module
@@ -85,9 +84,10 @@ def test_bs4_version_fails():
                          flavor='bs4')
 
 
-class TestReadHtml(unittest.TestCase):
+class TestReadHtml(tm.TestCase):
     @classmethod
     def setUpClass(cls):
+        super(TestReadHtml, cls).setupClass()
         _skip_if_none_of(('bs4', 'html5lib'))
 
     def read_html(self, *args, **kwargs):
@@ -582,9 +582,10 @@ def test_parse_dates_combine(self):
         tm.assert_frame_equal(newdf, res[0])
 
 
-class TestReadHtmlLxml(unittest.TestCase):
+class TestReadHtmlLxml(tm.TestCase):
     @classmethod
     def setUpClass(cls):
+        super(TestReadHtmlLxml, cls).setupClass()
         _skip_if_no('lxml')
 
     def read_html(self, *args, **kwargs):