def test_feature1(self):
import openmdao.api as om
from openmdao.core.tests.test_scaling import ScalingExample1
prob = om.Problem()
model = prob.model
model.add_subsystem('p1', om.IndepVarComp('x1', 1.0))
model.add_subsystem('p2', om.IndepVarComp('x2', 1.0))
comp = model.add_subsystem('comp', ScalingExample1())
model.connect('p1.x1', 'comp.x1')
model.connect('p2.x2', 'comp.x2')
prob.setup()
prob.run_model()
model.run_apply_nonlinear()
with model._scaled_context_all():
val = model.comp._outputs['y1']
assert_near_equal(val, 2.0)
val = model.comp._outputs['y2']
assert_near_equal(val, 6.0)
def test_feature1(self):
from openmdao.api import Problem, Group, IndepVarComp
from openmdao.core.tests.test_scaling import ScalingExample1
prob = Problem()
model = prob.model = Group()
model.add_subsystem('p1', IndepVarComp('x1', 1.0))
model.add_subsystem('p2', IndepVarComp('x2', 1.0))
comp = model.add_subsystem('comp', ScalingExample1())
model.connect('p1.x1', 'comp.x1')
model.connect('p2.x2', 'comp.x2')
prob.setup(check=False)
prob.run_model()
model.run_apply_nonlinear()
with model._scaled_context_all():
val = model.comp._outputs['y1']
assert_rel_error(self, val, 2.0)
val = model.comp._outputs['y2']
assert_rel_error(self, val, 6.0)