From 8cefde521122ce829060ae5e58db2202d9330020 Mon Sep 17 00:00:00 2001
From: Stefano Borini <sborini@enthought.com>
Date: Wed, 2 May 2018 17:37:53 +0100
Subject: [PATCH] Returns ProbeParameterFactory from ProbeMCOFactory
(cherry picked from commit 366edeb854acb34b05b5169f8c492dda7c433fe5)
---
force_bdss/tests/probe_classes/mco.py | 10 ++++++----
1 file changed, 6 insertions(+), 4 deletions(-)
diff --git a/force_bdss/tests/probe_classes/mco.py b/force_bdss/tests/probe_classes/mco.py
index 3690484..58fdd13 100644
--- a/force_bdss/tests/probe_classes/mco.py
+++ b/force_bdss/tests/probe_classes/mco.py
@@ -10,7 +10,11 @@ from force_bdss.api import (
class ProbeMCOModel(BaseMCOModel):
- pass
+ #: Counts how many times the edit_traits method has been called
+ edit_traits_call_count = Int(0)
+
+ def edit_traits(self, *args, **kwargs):
+ self.edit_traits_call_count += 1
def run_func(*args, **kwargs):
@@ -62,8 +66,6 @@ class ProbeMCOFactory(BaseMCOFactory):
mco_class = Type(ProbeMCO)
- probe_parameter_factories = List(Type(ProbeParameterFactory))
-
nb_output_data_values = Int(0)
def create_model(self, model_data=None):
@@ -83,4 +85,4 @@ class ProbeMCOFactory(BaseMCOFactory):
return self.mco_class(self)
def parameter_factories(self):
- return []
+ return [ProbeParameterFactory(mco_factory=self)]
--
GitLab