Skip to content
Snippets Groups Projects
Commit 68cec51e authored by Stefano Borini's avatar Stefano Borini
Browse files

Fixed tests

parent e622c1a5
No related branches found
No related tags found
1 merge request!45Added support for MCO named parameters.
......@@ -32,11 +32,12 @@ class BaseCoreDriver(Plugin):
def _parameter_factory_registry_default(self):
registry = MCOParameterFactoryRegistry()
for f in all_core_factories():
self.register(f)
registry.register(f)
return registry
def _workflow_default(self):
reader = WorkflowReader(self.bundle_registry)
reader = WorkflowReader(self.bundle_registry,
self.parameter_factory_registry)
with open(self.application.workflow_filepath) as f:
return reader.read(f)
......@@ -35,4 +35,3 @@ class TestDakotaCommunicator(unittest.TestCase):
self.assertEqual(len(data.value_names), 1)
self.assertEqual(len(data.value_types), 1)
self.assertEqual(len(data.values), 1)
......@@ -39,7 +39,6 @@ class WorkflowWriter(HasStrictTraits):
wf_data["multi_criteria_optimizer"]["model_data"]["parameters"] = parameters_data # noqa
kpic_data = []
for kpic in workflow.kpi_calculators:
kpic_data.append({
......
......@@ -26,5 +26,7 @@ def all_core_factories():
"""Produces a list of all factories contained in this module."""
import inspect
return [c for c in inspect.getmodule(all_core_factories).__dict__.values()
if inspect.isclass(c) and issubclass(c, BaseMCOParameterFactory)]
return [c() for c in inspect.getmodule(all_core_factories).__dict__.values()
if inspect.isclass(c) and
c is not BaseMCOParameterFactory and
issubclass(c, BaseMCOParameterFactory)]
......@@ -6,9 +6,9 @@ from force_bdss.mco.parameters.base_mco_parameter_factory import \
class TestCoreMCOParameters(unittest.TestCase):
def test_all_classes(self):
def test_all_factories(self):
factories = core_mco_parameters.all_core_factories()
self.assertNotEqual(len(factories), 0)
self.assertEqual(len(factories), 1)
for f in factories:
self.assertTrue(issubclass(f, BaseMCOParameterFactory))
self.assertIsInstance(f, BaseMCOParameterFactory)
......@@ -2,13 +2,14 @@
"version": "1",
"workflow": {
"multi_criteria_optimizer": {
"id": "force.bdss.bundles.enthought.dakota",
"id": "force.bdss.bundle.enthought.dakota",
"model_data": {
"parameters" : []
}
},
"data_sources": [
{
"id": "force.bdss.bundles.enthought.csv_extractor",
"id": "force.bdss.bundle.enthought.csv_extractor",
"model_data": {
"filename": "foo.csv",
"row": 3,
......@@ -17,7 +18,7 @@
}
},
{
"id": "force.bdss.bundles.enthought.csv_extractor",
"id": "force.bdss.bundle.enthought.csv_extractor",
"model_data": {
"filename": "foo.csv",
"row": 3,
......@@ -28,7 +29,7 @@
],
"kpi_calculators": [
{
"id": "force.bdss.bundles.enthought.kpi_adder",
"id": "force.bdss.bundle.enthought.kpi_adder",
"model_data": {
"cuba_type_in": "PRESSURE",
"cuba_type_out": "TOTAL_PRESSURE"
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment