Move init class for the factory

force_bdss/tests/probe_classes.py

@@ -41,6 +41,13 @@ class ProbeEvaluator(HasStrictTraits):
 
 
 class ProbeEvaluatorFactory(HasStrictTraits):
+    def __init__(self, plugin=None, *args, **kwargs):
+        if plugin is None:
+            plugin = mock.Mock(Plugin)
+
+        super(ProbeEvaluatorFactory, self).__init__(
+            plugin=plugin, *args, **kwargs)
+
     run_function = Function
 
     input_slots_type = Str('PRESSURE')
@@ -70,13 +77,6 @@ class ProbeKPICalculatorFactory(BaseKPICalculatorFactory,
     id = Str('enthought.test.datasource')
     name = Str('test_datasource')
 
-    def __init__(self, plugin=None, *args, **kwargs):
-        if plugin is None:
-            plugin = mock.Mock(Plugin)
-
-        super(ProbeKPICalculatorFactory, self).__init__(
-            plugin=plugin, *args, **kwargs)
-
     def create_model(self, model_data=None):
         return ProbeKPICalculatorModel(self, self.model_data)