From dbdf6840a0a288f7bd667261a0abaf576d9c2020 Mon Sep 17 00:00:00 2001
From: martinRenou <martin.renou@gmail.com>
Date: Thu, 24 Aug 2017 12:00:03 +0100
Subject: [PATCH] Move init class for the factory
---
force_bdss/tests/probe_classes.py | 14 +++++++-------
1 file changed, 7 insertions(+), 7 deletions(-)
diff --git a/force_bdss/tests/probe_classes.py b/force_bdss/tests/probe_classes.py
index b9b54da..54d7b6c 100644
--- a/force_bdss/tests/probe_classes.py
+++ b/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)
--
GitLab