Adding shim compatability for python2.7 and python3.6 by allowing for…

… assertCountEqual in both

tests/testautomater.py

@@ -4,11 +4,12 @@
 
 from keras_pandas import lib
 from keras_pandas.Automater import Automater
+from tests.testbase import TestBase
 
 logging.getLogger().setLevel(logging.INFO)
 
 
-class TestAutomater(unittest.TestCase):
+class TestAutomater(TestBase):
 
     def test_check_variable_lists_are_valid(self):
         # Base case: No variables
@@ -104,7 +105,6 @@ def test_initializer(self):
 
         response_variable_list = [item for sublist in response.values() for item in sublist]
         self.assertCountEqual(response_variable_list, auto._user_provided_variables)
-
         # Overlapping variable lists
         data = {
             'numerical_vars': ['n1', 'n2', 'n3', 'x1'],

tests/testbase.py

@@ -0,0 +1,11 @@
+import unittest
+
+
+class TestBase(unittest.TestCase):
+
+    def __init__(self, *args, **kwargs):
+        super(TestBase, self).__init__(*args, **kwargs)
+
+        if not hasattr(self, 'assertCountEqual'):
+            self.assertCountEqual = self.assertItemsEqual
+

tests/testcategorical.py

@@ -8,9 +8,10 @@
 
 from keras_pandas import lib
 from keras_pandas.Automater import Automater
+from tests.testbase import TestBase
 
 
-class TestCategorical(unittest.TestCase):
+class TestCategorical(TestBase):
 
     def test_fit(self):
         train_df = lib.load_mushroom()

tests/testembeddingvectorizer.py

@@ -6,11 +6,12 @@
 
 from keras_pandas import lib
 from keras_pandas.transformations import EmbeddingVectorizer
+from tests.testbase import TestBase
 
 logging.getLogger().setLevel(logging.INFO)
 
 
-class TestEmbeddingVectorizer(unittest.TestCase):
+class TestEmbeddingVectorizer(TestBase):
 
     def test_padding(self):
         # Empty string

tests/testnumeric.py

@@ -6,11 +6,12 @@
 
 from keras_pandas import lib
 from keras_pandas.Automater import Automater
+from tests.testbase import TestBase
 
 logging.getLogger().setLevel(logging.INFO)
 
 
-class TestNumeric(unittest.TestCase):
+class TestNumeric(TestBase):
 
     def test_fit(self):
         iris_df = lib.load_iris()

tests/testtext.py

@@ -8,11 +8,12 @@
 
 from keras_pandas import lib
 from keras_pandas.Automater import Automater
+from tests.testbase import TestBase
 
 logging.getLogger().setLevel(logging.INFO)
 
 
-class TestText(unittest.TestCase):
+class TestText(TestBase):
 
     def test_fit(self):
         data = lib.load_titanic()