Fix test code style

Author: anatolfernandez

test/automl/test_models.py

@@ -6,7 +6,7 @@
 from autosklearn.util.backend import Backend
 
 
-class AutoMLStub(object):
+class AutoMLStub(AutoML):
 
     def __init__(self):
         self.__class__ = AutoML
@@ -46,4 +46,4 @@ def test_raises_if_no_models(self):
         self.automl._backend.load_ensemble.return_value = None
         self.automl._backend.load_all_models.return_value = []
 
-        self.assertRaises(ValueError, self.automl._load_models)
+        self.assertRaises(ValueError, self.automl._load_models)

test/util/test_backend.py

@@ -9,14 +9,13 @@ class BackendModelsTest(unittest.TestCase):
 
     class BackendStub(Backend):
 
-        def __init__(self, model_directory):
+        def __init__(self):
             self.__class__ = Backend
-            self.get_model_dir = lambda: model_directory
 
     def setUp(self):
         self.model_directory = '/model_directory/'
-        self.backend = self.BackendStub(self.model_directory)
-        self.BackendStub.get_model_dir = lambda x: 42
+        self.backend = self.BackendStub()
+        self.backend.get_model_dir = lambda: self.model_directory
 
     @mock.patch('six.moves.cPickle.load')
     @mock.patch('__builtin__.open')
@@ -53,4 +52,4 @@ def _setup_load_model_mocks(self, openMock, pickleLoadMock, seed, idx):
         openMock.side_effect = lambda path, flag: fileMock if path == model_path and flag == 'rb' else None
         pickleLoadMock.side_effect = lambda fh: expected_model if fh == file_handler else None
 
-        return expected_model
+        return expected_model