def setUp(self):
from openmdao.api import Group, Problem, IndepVarComp
from openmdao.core.tests.test_impl_comp import QuadraticComp
group = Group()
comp1 = group.add_subsystem('comp1', IndepVarComp())
comp1.add_output('a', 1.0)
comp1.add_output('b', 1.0)
comp1.add_output('c', 1.0)
sub = group.add_subsystem('sub', Group())
sub.add_subsystem('comp2', QuadraticComp())
sub.add_subsystem('comp3', QuadraticComp())
group.connect('comp1.a', 'sub.comp2.a')
group.connect('comp1.b', 'sub.comp2.b')
group.connect('comp1.c', 'sub.comp2.c')
group.connect('comp1.a', 'sub.comp3.a')
group.connect('comp1.b', 'sub.comp3.b')
group.connect('comp1.c', 'sub.comp3.c')
prob = Problem(model=group)
prob.setup()
prob['comp1.a'] = 1.
prob['comp1.b'] = -4.
prob['comp1.c'] = 3.
prob.run_model()
def setup(self):
comp1 = self.add_subsystem('comp1', IndepVarComp())
comp1.add_output('a', 1.0)
comp1.add_output('b', 1.0)
comp1.add_output('c', 1.0)
sub = self.add_subsystem('sub', Group())
sub.add_subsystem('comp2', QuadraticComp())
sub.add_subsystem('comp3', QuadraticComp())
self.connect('comp1.a', 'sub.comp2.a')
self.connect('comp1.b', 'sub.comp2.b')
self.connect('comp1.c', 'sub.comp2.c')
self.connect('comp1.a', 'sub.comp3.a')
self.connect('comp1.b', 'sub.comp3.b')
self.connect('comp1.c', 'sub.comp3.c')
def test_simple_list_vars_options(self):
from openmdao.api import Group, Problem, IndepVarComp
class QuadraticComp(ImplicitComponent):
"""
A Simple Implicit Component representing a Quadratic Equation.
R(a, b, c, x) = ax^2 + bx + c
Solution via Quadratic Formula:
x = (-b + sqrt(b^2 - 4ac)) / 2a
"""
def setup(self):
self.add_input('a', val=1., units='ft')
self.add_input('b', val=1., units='inch')
self.add_input('c', val=1., units='ft')
self.add_output('x',
val=0.,
lower=1.0,
upper=100.0,
ref=1.1,
ref0=2.1,
units='inch')
self.declare_partials(of='*', wrt='*')
def apply_nonlinear(self, inputs, outputs, residuals):
a = inputs['a']
b = inputs['b']
c = inputs['c']
x = outputs['x']
residuals['x'] = a * x**2 + b * x + c
def solve_nonlinear(self, inputs, outputs):
a = inputs['a']
b = inputs['b']
c = inputs['c']
outputs['x'] = (-b + (b**2 - 4 * a * c)**0.5) / (2 * a)
group = Group()
comp1 = group.add_subsystem('comp1', IndepVarComp())
comp1.add_output('a', 1.0, units='ft')
comp1.add_output('b', 1.0, units='inch')
comp1.add_output('c', 1.0, units='ft')
sub = group.add_subsystem('sub', Group())
sub.add_subsystem('comp2', QuadraticComp())
sub.add_subsystem('comp3', QuadraticComp())
group.connect('comp1.a', 'sub.comp2.a')
group.connect('comp1.b', 'sub.comp2.b')
group.connect('comp1.c', 'sub.comp2.c')
group.connect('comp1.a', 'sub.comp3.a')
group.connect('comp1.b', 'sub.comp3.b')
group.connect('comp1.c', 'sub.comp3.c')
global prob
prob = Problem(model=group)
prob.setup()
prob['comp1.a'] = 1.
prob['comp1.b'] = -4.
prob['comp1.c'] = 3.
prob.run_model()
# list_inputs test
stream = cStringIO()
inputs = prob.model.list_inputs(values=False, out_stream=stream)
text = stream.getvalue()
self.assertEqual(sorted(inputs), [
('sub.comp2.a', {}),
('sub.comp2.b', {}),
('sub.comp2.c', {}),
('sub.comp3.a', {}),
('sub.comp3.b', {}),
('sub.comp3.c', {}),
])
self.assertEqual(1, text.count("6 Input(s) in 'model'"))
self.assertEqual(1, text.count("top"))
self.assertEqual(1, text.count(" sub"))
self.assertEqual(1, text.count(" comp2"))
self.assertEqual(2, text.count(" a"))
num_non_empty_lines = sum([1 for s in text.splitlines() if s.strip()])
self.assertEqual(num_non_empty_lines, 14)
# list_outputs tests
# list implicit outputs
outputs = prob.model.list_outputs(explicit=False, out_stream=None)
text = stream.getvalue()
self.assertEqual(sorted(outputs), [('sub.comp2.x', {
'value': [3.]
}), ('sub.comp3.x', {
'value': [3.]
})])
# list explicit outputs
stream = cStringIO()
outputs = prob.model.list_outputs(implicit=False, out_stream=None)
self.assertEqual(sorted(outputs), [
('comp1.a', {
'value': [1.]
}),
('comp1.b', {
'value': [-4.]
}),
('comp1.c', {
'value': [3.]
}),
])
