def test_nested_2Darray_simul_element_and_full_connection(self):
top = Assembly()
top.add('comp', ArrayComp2D())
top.add('nest', Assembly())
top.nest.add('comp', ArrayComp2D())
top.driver.gradient_options.lin_solver = 'petsc_ksp'
top.nest.driver.gradient_options.lin_solver = 'petsc_ksp'
top.nest.driver.workflow.add(['comp'])
top.nest.create_passthrough('comp.x')
top.nest.create_passthrough('comp.y')
top.add('driver', SimpleDriver())
top.driver.workflow.add(['nest', 'comp'])
top.connect('nest.y', 'comp.x')
top.driver.add_parameter('nest.x[0][0]', low=-10, high=10)
top.driver.add_objective('comp.y[0][0]')
top.driver.add_constraint('nest.y[0][1] < 0')
top.run()
J = top.driver.calc_gradient(mode='forward')
assert_rel_error(self, J[0, 0], 24.0, .000001)
assert_rel_error(self, J[1, 0], 4.0, .000001)
J = top.driver.calc_gradient(mode='adjoint')
assert_rel_error(self, J[0, 0], 24.0, .000001)
assert_rel_error(self, J[1, 0], 4.0, .000001)
J = top.driver.calc_gradient(mode='fd')
assert_rel_error(self, J[0, 0], 24.0, .000001)
assert_rel_error(self, J[1, 0], 4.0, .000001)
def test_linearGS_simul_element_and_full_connection(self):
# Added because of a bug with array slices for Linear GS
top = Assembly()
top.add('comp1', ArrayComp2D())
top.add('comp2', ArrayComp2D())
top.add('driver', SimpleDriver())
top.driver.workflow.add(['comp1', 'comp2'])
top.connect('comp1.y', 'comp2.x')
top.driver.add_parameter('comp1.x[0][0]', low=-10, high=10)
top.driver.add_objective('comp1.y[0][0]')
top.driver.add_constraint('comp2.y[0][1] < 0')
top.driver.gradient_options.lin_solver = 'linear_gs'
top.driver.gradient_options.maxiter = 1
top.run()
J = top.driver.calc_gradient(mode='forward')
assert_rel_error(self, J[0, 0], 2.0, .000001)
assert_rel_error(self, J[1, 0], 39.0, .000001)
J = top.driver.calc_gradient(mode='adjoint')
assert_rel_error(self, J[0, 0], 2.0, .000001)
assert_rel_error(self, J[1, 0], 39.0, .000001)
def test_linearGS_simul_element_and_full_connection(self):
# Added because of a bug with array slices for Linear GS
top = Assembly()
top.add('comp1', ArrayComp2D())
top.add('comp2', ArrayComp2D())
top.add('driver', SimpleDriver())
top.driver.workflow.add(['comp1', 'comp2'])
top.connect('comp1.y', 'comp2.x')
top.driver.add_parameter('comp1.x[0][0]', low=-10, high=10)
top.driver.add_objective('comp1.y[0][0]')
top.driver.add_constraint('comp2.y[0][1] < 0')
top.driver.gradient_options.lin_solver = 'linear_gs'
top.driver.gradient_options.maxiter = 1
top.run()
J = top.driver.calc_gradient(mode='forward')
assert_rel_error(self, J[0, 0], 2.0, .000001)
assert_rel_error(self, J[1, 0], 39.0, .000001)
J = top.driver.calc_gradient(mode='adjoint')
assert_rel_error(self, J[0, 0], 2.0, .000001)
assert_rel_error(self, J[1, 0], 39.0, .000001)
class GeomTestCase(unittest.TestCase):
def setUp(self):
"""this setup function will be called before each test in this class"""
self.top = Assembly()
self.top.add('g1', GeoSource())
self.top.add('g2', GeoSink())
self.top.connect('g1.g_out', 'g2.g_inp')
self.top.driver.workflow.add(['g1', 'g2'])
def tearDown(self):
"""this teardown function will be called after each test"""
self.top = None
def testGeom(self):
self.top.run()
self.assertEqual(1, self.top.g2.g_inp.get_tessellation())
try:
self.top.g2.g_extra = "hey"
except TypeError as err:
msg = "g2 (1-g2): object g_extra of type <type 'str'> must provide interface 'IStaticGeometry'"
self.assertEqual(str(err), msg)
else:
self.fail("exception expected")
def test_nested_2Darray_simul_element_and_full_connection(self):
top = Assembly()
top.add('comp', ArrayComp2D())
top.add('nest', Assembly())
top.nest.add('comp', ArrayComp2D())
top.driver.gradient_options.lin_solver = 'petsc_ksp'
top.nest.driver.gradient_options.lin_solver = 'petsc_ksp'
top.nest.driver.workflow.add(['comp'])
top.nest.create_passthrough('comp.x')
top.nest.create_passthrough('comp.y')
top.add('driver', SimpleDriver())
top.driver.workflow.add(['nest', 'comp'])
top.connect('nest.y', 'comp.x')
top.driver.add_parameter('nest.x[0][0]', low=-10, high=10)
top.driver.add_objective('comp.y[0][0]')
top.driver.add_constraint('nest.y[0][1] < 0')
top.run()
J = top.driver.calc_gradient(mode='forward')
assert_rel_error(self, J[0, 0], 24.0, .000001)
assert_rel_error(self, J[1, 0], 4.0, .000001)
J = top.driver.calc_gradient(mode='adjoint')
assert_rel_error(self, J[0, 0], 24.0, .000001)
assert_rel_error(self, J[1, 0], 4.0, .000001)
J = top.driver.calc_gradient(mode='fd')
assert_rel_error(self, J[0, 0], 24.0, .000001)
assert_rel_error(self, J[1, 0], 4.0, .000001)
class GeomTestCase(unittest.TestCase):
def setUp(self):
"""this setup function will be called before each test in this class"""
self.top = Assembly()
self.top.add('g1', GeoSource())
self.top.add('g2', GeoSink())
self.top.connect('g1.g_out', 'g2.g_inp')
self.top.driver.workflow.add(['g1', 'g2'])
def tearDown(self):
"""this teardown function will be called after each test"""
self.top = None
def testGeom(self):
self.top.run()
self.assertEqual(1, self.top.g2.g_inp.get_tessellation())
try:
self.top.g2.g_extra = "hey"
except TypeError as err:
msg = "g2 (1-2): g_extra must provide interface 'IStaticGeometry'"
self.assertEqual(str(err), msg)
else:
self.fail("exception expected")
def _create_assembly(self, dbname):
asm = Assembly()
driver = asm.add('driver', SimpleCaseIterDriver())
asm.add('comp1', ExecComp(exprs=['z=x+y']))
asm.add('comp2', ExecComp(exprs=['z=x+y']))
asm.connect('comp1.z', 'comp2.x')
driver.workflow.add(['comp1', 'comp2'])
driver.recorder = DBCaseRecorder(dbname, append=True)
return asm
def _create_assembly(self, dbname, drivertype):
asm = Assembly()
driver = asm.add('driver', drivertype())
asm.add('comp1', ExecComp(exprs=['z=x+y']))
asm.add('comp2', ExecComp(exprs=['z=x+y']))
asm.connect('comp1.z', 'comp2.x')
driver.workflow.add(['comp1', 'comp2'])
driver.recorders = [DBCaseRecorder(dbname, append=True)]
return asm
def _create_assembly(self, dbname, drivertype):
asm = Assembly()
driver = asm.add('driver', drivertype())
asm.add('comp1', TracedExecComp(exprs=['z=x+y']))
asm.add('comp2', TracedExecComp(exprs=['z=x+y']))
asm.connect('comp1.z', 'comp2.x')
driver.workflow.add(['comp1', 'comp2'])
asm.recorders = [DBCaseRecorder(dbname, append=True)]
return asm
def _create_assembly(self, dbname, drivertype):
asm = Assembly()
driver = asm.add("driver", drivertype())
asm.add("comp1", ExecComp(exprs=["z=x+y"]))
asm.add("comp2", ExecComp(exprs=["z=x+y"]))
asm.connect("comp1.z", "comp2.x")
driver.workflow.add(["comp1", "comp2"])
driver.recorders = [DBCaseRecorder(dbname, append=True)]
return asm
def test_multiconnect(self):
top = Assembly()
for name in ("m1", "m2", "m3"):
top.add(name, Multiplier())
top.driver.workflow.add(name)
top.connect("m1.rval_out", ("m2.mult", "m3.mult"))
top.m1.rval_in = 1.0
top.m2.rval_in = 3.0
top.m3.rval_in = 4.0
top.run()
self.assertEqual(top.m2.rval_out, 4.5)
self.assertEqual(top.m3.rval_out, 6.0)
def test_multiconnect(self):
top = Assembly()
for name in ('m1', 'm2', 'm3'):
top.add(name, Multiplier())
top.driver.workflow.add(name)
top.connect('m1.rval_out', ('m2.mult', 'm3.mult'))
top.m1.rval_in = 1.
top.m2.rval_in = 3.
top.m3.rval_in = 4.
top.run()
self.assertEqual(top.m2.rval_out, 4.5)
self.assertEqual(top.m3.rval_out, 6.)
def test_multiconnect(self):
top = Assembly()
for name in ('m1', 'm2', 'm3'):
top.add(name, Multiplier())
top.driver.workflow.add(name)
top.connect('m1.rval_out', ('m2.mult', 'm3.mult'))
top.m1.rval_in = 1.
top.m2.rval_in = 3.
top.m3.rval_in = 4.
top.run()
self.assertEqual(top.m2.rval_out, 4.5)
self.assertEqual(top.m3.rval_out, 6.)
def test_scid2(self):
logging.debug('')
logging.debug('test_simplecid')
top = Assembly()
top.add('generator', Generator())
cid = top.add('cid', SimpleCaseIterDriver())
top.add('driven', DrivenComponent())
top.add('verifier', Verifier())
top.driver.workflow.add(('generator', 'cid', 'verifier'))
cid.workflow.add('driven')
cid.add_parameter('driven.x')
cid.add_parameter('driven.y')
cid.add_response('driven.rosen_suzuki')
cid.add_response('driven.sum_y')
top.connect('generator.x', 'cid.case_inputs.driven.x')
top.connect('generator.y', 'cid.case_inputs.driven.y')
top.connect('generator.x', 'verifier.x')
top.connect('generator.y', 'verifier.y')
top.connect('cid.case_outputs.driven.rosen_suzuki', 'verifier.rosen_suzuki')
top.connect('cid.case_outputs.driven.sum_y', 'verifier.sum_y')
top.run()
def test_connections(self):
asm = Assembly()
asm.add('dummy1',Dummy())
asm.add('dummy2',Dummy())
asm.add('bcast',Broadcaster(['x']))
asm.connect('dummy1.x','bcast.x_in')
asm.connect('bcast.x','dummy2.y')
self.assertEqual(set(asm.list_connections()),set([('dummy1.x', 'bcast.x_in'), ('bcast.x', 'dummy2.y')]))
asm.bcast.names = ['z']
self.assertEqual(asm.list_connections(),[])
def test_scid2(self):
logging.debug('')
logging.debug('test_simplecid')
top = Assembly()
top.add('generator', Generator())
cid = top.add('cid', SimpleCaseIterDriver())
top.add('driven', DrivenComponent())
top.add('verifier', Verifier())
top.driver.workflow.add(('generator', 'cid', 'verifier'))
cid.workflow.add('driven')
cid.add_parameter('driven.x')
cid.add_parameter('driven.y')
cid.add_response('driven.rosen_suzuki')
cid.add_response('driven.sum_y')
top.connect('generator.x', 'cid.case_inputs.driven.x')
top.connect('generator.y', 'cid.case_inputs.driven.y')
top.connect('generator.x', 'verifier.x')
top.connect('generator.y', 'verifier.y')
top.connect('cid.case_outputs.driven.rosen_suzuki',
'verifier.rosen_suzuki')
top.connect('cid.case_outputs.driven.sum_y', 'verifier.sum_y')
top.run()
def test_connections(self):
logging.debug('')
logging.debug('test_connections')
top = Assembly()
top.add('generator', Generator())
top.add('cid', CaseIteratorDriver())
top.add('driven', DrivenComponent())
top.add('verifier', Verifier())
top.driver.workflow.add(('generator', 'cid', 'verifier'))
top.cid.workflow.add('driven')
top.connect('generator.cases', 'cid.iterator')
top.connect('cid.evaluated', 'verifier.cases')
top.run()
def test_connections(self):
logging.debug("")
logging.debug("test_connections")
top = Assembly()
top.add("generator", Generator())
top.add("cid", CaseIteratorDriver())
top.add("driven", DrivenComponent())
top.add("verifier", Verifier())
top.driver.workflow.add(("generator", "cid", "verifier"))
top.cid.workflow.add("driven")
top.cid.printvars = ["driven.extra"]
top.connect("generator.cases", "cid.iterator")
top.connect("cid.evaluated", "verifier.cases")
top.run()
def test_connections(self):
logging.debug('')
logging.debug('test_connections')
top = Assembly()
top.add('generator', Generator())
top.add('cid', CaseIteratorDriver())
top.add('driven', DrivenComponent())
top.add('verifier', Verifier())
top.driver.workflow.add(('generator', 'cid', 'verifier'))
top.cid.workflow.add('driven')
top.cid.printvars = ['driven.extra']
top.connect('generator.cases', 'cid.iterator')
top.connect('cid.evaluated', 'verifier.cases')
top.run()
def test_nested(self):
asm3 = Assembly()
asm3.add('comp1', ExecComp(exprs=['z=x+y']))
driver = asm3.add('driver', SLSQPdriver())
driver.workflow.add('comp1')
driver.add_parameter('comp1.y', low=-1, high=1, start=0)
driver.add_objective('comp1.z')
driver.add_constraint('comp1.z >= 0')
asm3.create_passthrough('comp1.x')
asm3.create_passthrough('comp1.z')
asm2 = Assembly()
asm2.add('comp1', ExecComp(exprs=['z=x+y']))
asm2.add('asm3', asm3)
asm2.connect('comp1.z', 'asm3.x')
driver = asm2.add('driver', SLSQPdriver())
driver.workflow.add(('comp1', 'asm3'))
driver.add_parameter('comp1.y', low=-1, high=1, start=0)
driver.add_objective('asm3.z')
driver.add_constraint('comp1.z >= 0')
asm2.create_passthrough('comp1.x')
asm2.create_passthrough('asm3.z')
asm1 = set_as_top(Assembly())
asm1.add('comp1', ExecComp(exprs=['z=x+y']))
asm1.add('asm2', asm2)
asm1.connect('comp1.z', 'asm2.x')
driver = asm1.add('driver', SLSQPdriver())
driver.workflow.add(('comp1', 'asm2'))
driver.add_parameter('comp1.y', low=-1, high=1, start=0)
driver.add_objective('asm2.z')
driver.add_constraint('comp1.z >= 0')
sout = StringIO()
asm1.recorders = [JSONCaseRecorder(sout)]
asm1.run()
if os.environ.get('REGEN_JSON_FILES'):
with open('nested.new', 'w') as out:
out.write(sout.getvalue())
verify_json(self, sout, 'nested.json')
def test_nested(self):
asm3 = Assembly()
asm3.add('comp1', ExecComp(exprs=['z=x+y']))
driver = asm3.add('driver', SLSQPdriver())
driver.workflow.add('comp1')
driver.add_parameter('comp1.y', low=-1, high=1, start=0)
driver.add_objective('comp1.z')
driver.add_constraint('comp1.z >= 0')
asm3.create_passthrough('comp1.x')
asm3.create_passthrough('comp1.z')
asm2 = Assembly()
asm2.add('comp1', ExecComp(exprs=['z=x+y']))
asm2.add('asm3', asm3)
asm2.connect('comp1.z', 'asm3.x')
driver = asm2.add('driver', SLSQPdriver())
driver.workflow.add(('comp1', 'asm3'))
driver.add_parameter('comp1.y', low=-1, high=1, start=0)
driver.add_objective('asm3.z')
driver.add_constraint('comp1.z >= 0')
asm2.create_passthrough('comp1.x')
asm2.create_passthrough('asm3.z')
asm1 = set_as_top(Assembly())
asm1.add('comp1', ExecComp(exprs=['z=x+y']))
asm1.add('asm2', asm2)
asm1.connect('comp1.z', 'asm2.x')
driver = asm1.add('driver', SLSQPdriver())
driver.workflow.add(('comp1', 'asm2'))
driver.add_parameter('comp1.y', low=-1, high=1, start=0)
driver.add_objective('asm2.z')
driver.add_constraint('comp1.z >= 0')
sout = StringIO()
asm1.recorders = [JSONCaseRecorder(sout)]
asm1.run()
if os.environ.get('REGEN_JSON_FILES'):
with open('nested.new', 'w') as out:
out.write(sout.getvalue())
verify_json(self, sout, 'nested.json')
def execute(self):
""" Calculate the wingsurface from the spar geometry """
# Make sure that input data has the same length
self.verify_input()
for beam in self.eqbeams:
a = Assembly()
# Add a splined blade planform from the input data
a.add('pf_splines', SplinedBladePlanform(True))
a.driver.workflow.add('pf_splines')
a.pf_splines.nC = 3
a.pf_splines.pfIn = beam
a.pf_splines.configure_splines()
a.create_passthrough('pf_splines.blade_length')
a.create_passthrough('pf_splines.span_ni')
a.add('blade_surface', LoftedBladeSurface())
a.driver.workflow.add('blade_surface')
a.connect('pf_splines.pfOut', 'blade_surface.pf')
a.connect('span_ni', 'blade_surface.span_ni')
# load the planform file
a.span_ni = self.span_ni
b = a.blade_surface
# distribute 200 points evenly along the airfoil sections
b.chord_ni = 200
for f in self.airfoils:
b.base_airfoils.append(np.loadtxt(f))
b.blend_var = np.array([0.25, 0.75])
self.wingsurf.append(a)
class TestCase_Residuals(unittest.TestCase):
""" Test run/step/stop aspects of a simple workflow. """
def setUp(self):
""" Called before each test. """
self.model = Assembly()
self.model.add('c1', MyComp())
self.model.add('c2', MyComp())
self.model.add('driver', SimpleDriver())
self.model.driver.workflow = CyclicWorkflow()
self.model.driver.workflow.add(['c1', 'c2'])
def tearDown(self):
""" Called after each test. """
self.model = None
def test_column_vector(self):
self.model.c1.add('y_a', Array(iotype='out'))
self.model.c1.y_a = array([[1.0], [2.0]])
self.model.c2.add('x_a', Array(iotype='in'))
self.model.c2.x_a = array([[3.0], [6.0]])
self.model.connect('c1.y_a', 'c2.x_a')
self.model.connect('c2.y', 'c1.x')
self.model.driver.workflow.initialize_residual()
self.model.run()
indep = self.model.driver.workflow.get_independents()
self.assertEqual(indep[0], 1.0)
self.assertEqual(indep[1], 2.0)
dep = self.model.driver.workflow.get_dependents()
self.assertEqual(dep[0], 0.0)
self.assertEqual(dep[1], 0.0)
dv = array([3.0, 5.0])
self.model.driver.workflow.set_independents(dv)
indep = self.model.driver.workflow.get_independents()
self.assertEqual(indep[0], 3.0)
self.assertEqual(indep[1], 5.0)
def test_row_vector(self):
self.model.c1.add('y_a', Array(iotype='out'))
self.model.c1.y_a = array([[1.0, 2.0]])
self.model.c2.add('x_a', Array(iotype='in'))
self.model.c2.x_a = array([[3.0, 6.0]])
self.model.connect('c1.y_a', 'c2.x_a')
self.model.connect('c2.y', 'c1.x')
self.model.driver.workflow.initialize_residual()
self.model.run()
indep = self.model.driver.workflow.get_independents()
self.assertEqual(indep[0], 1.0)
self.assertEqual(indep[1], 2.0)
dep = self.model.driver.workflow.get_dependents()
self.assertEqual(dep[0], 0.0)
self.assertEqual(dep[1], 0.0)
dv = array([3.0, 5.0])
self.model.driver.workflow.set_independents(dv)
indep = self.model.driver.workflow.get_independents()
self.assertEqual(indep[0], 3.0)
self.assertEqual(indep[1], 5.0)
def test_array_1D(self):
self.model.c1.add('y_a', Array(iotype='out'))
self.model.c1.y_a = array([1.0, 2.0])
self.model.c2.add('x_a', Array(iotype='in'))
self.model.c2.x_a = array([3.0, 6.0])
self.model.connect('c1.y_a', 'c2.x_a')
self.model.connect('c2.y', 'c1.x')
self.model.driver.workflow.initialize_residual()
self.model.run()
indep = self.model.driver.workflow.get_independents()
self.assertEqual(indep[0], 1.0)
self.assertEqual(indep[1], 2.0)
dep = self.model.driver.workflow.get_dependents()
self.assertEqual(dep[0], 0.0)
self.assertEqual(dep[1], 0.0)
dv = array([3.0, 5.0])
self.model.driver.workflow.set_independents(dv)
indep = self.model.driver.workflow.get_independents()
self.assertEqual(indep[0], 3.0)
self.assertEqual(indep[1], 5.0)
def test_full_matrix(self):
self.model.c1.add('y_a', Array(iotype='out'))
self.model.c1.y_a = array([[1.0, 2.0], [3.0, 4.0]])
self.model.c2.add('x_a', Array(iotype='in'))
self.model.c2.x_a = array([[2.0, 5.0], [11.0, 17.0]])
self.model.connect('c1.y_a', 'c2.x_a')
self.model.connect('c2.y', 'c1.x')
self.model.driver.workflow.initialize_residual()
self.model.run()
indep = self.model.driver.workflow.get_independents()
self.assertEqual(indep[0], 1.0)
self.assertEqual(indep[1], 2.0)
self.assertEqual(indep[2], 3.0)
self.assertEqual(indep[3], 4.0)
dep = self.model.driver.workflow.get_dependents()
self.assertEqual(dep[0], 0.0)
self.assertEqual(dep[1], 0.0)
self.assertEqual(dep[2], 0.0)
self.assertEqual(dep[3], 0.0)
dv = array([3.0, 5.0, -8.0, -13.0])
self.model.driver.workflow.set_independents(dv)
indep = self.model.driver.workflow.get_independents()
self.assertEqual(indep[0], 3.0)
self.assertEqual(indep[1], 5.0)
self.assertEqual(indep[2], -8.0)
self.assertEqual(indep[3], -13.0)
def test_matrix_element(self):
# Array element to scalar.
self.model.c1.add('y_a', Array(iotype='out'))
self.model.c1.y_a = array([[1.0, 2.0], [3.0, 4.0]])
self.model.c2.x = 7.0
self.model.connect('c1.y_a[0, 0]', 'c2.x')
self.model.connect('c2.y', 'c1.x')
self.model.driver.workflow.initialize_residual()
self.model.run()
indep = self.model.driver.workflow.get_independents()
self.assertEqual(indep[0], 1.0)
dep = self.model.driver.workflow.get_dependents()
self.assertEqual(dep[0], 0.0)
dv = array([35.0])
self.model.driver.workflow.set_independents(dv)
indep = self.model.driver.workflow.get_independents()
self.assertEqual(indep[0], 35.0)
def test_vtree(self):
self.model.c1.add('vt_out', VarTree(Tree1(), iotype='out'))
self.model.c2.add('vt_in', VarTree(Tree1(), iotype='in'))
self.model.c2.vt_in.a1 = 4.
self.model.c2.vt_in.b1 = array([7.0, 12.0])
self.model.c2.vt_in.vt1.c1 = 13.
self.model.c2.vt_in.vt1.d1 = array([-1.0, 2.0])
self.model.connect('c1.vt_out', 'c2.vt_in')
self.model.connect('c2.y', 'c1.x')
self.model.driver.workflow.initialize_residual()
indep = self.model.driver.workflow.get_independents()
self.assertEqual(indep[0], 4.0)
self.assertEqual(indep[1], 7.0)
self.assertEqual(indep[2], 12.0)
self.assertEqual(indep[3], 13.0)
self.assertEqual(indep[4], -1.0)
self.assertEqual(indep[5], 2.0)
dep = self.model.driver.workflow.get_dependents()
self.assertEqual(dep[0], -1.0)
self.assertEqual(dep[1], -4.0)
self.assertEqual(dep[2], -5.0)
self.assertEqual(dep[3], -4.0)
self.assertEqual(dep[4], 6.0)
self.assertEqual(dep[5], 9.0)
dv = array([81.0, 82.0, 83.0, 84.0, 85.0, 86.0])
self.model.driver.workflow.set_independents(dv)
indep = self.model.driver.workflow.get_independents()
self.assertEqual(indep[0], 81.0)
self.assertEqual(indep[1], 82.0)
self.assertEqual(indep[2], 83.0)
self.assertEqual(indep[3], 84.0)
self.assertEqual(indep[4], 85.0)
self.assertEqual(indep[5], 86.0)
def test_vtree_leaf(self):
self.model.c1.add('vt_out', VarTree(Tree1(), iotype='out'))
self.model.c2.add('vt_in', VarTree(Tree1(), iotype='in'))
self.model.c2.vt_in.b1 = array([7.0, 12.0])
self.model.connect('c1.vt_out.b1', 'c2.vt_in.b1')
self.model.connect('c2.y', 'c1.x')
self.model.driver.workflow.initialize_residual()
indep = self.model.driver.workflow.get_independents()
self.assertEqual(indep[0], 7.0)
self.assertEqual(indep[1], 12.0)
dep = self.model.driver.workflow.get_dependents()
self.assertEqual(dep[0], -4.0)
self.assertEqual(dep[1], -5.0)
dv = array([13.0, 15.0])
self.model.driver.workflow.set_independents(dv)
indep = self.model.driver.workflow.get_independents()
self.assertEqual(indep[0], 13.0)
self.assertEqual(indep[1], 15.0)
wt.cone_angle = -2.5
wt.hub_radius = 2.8
wt.blade_length = 86.366
wt.rotor_diameter = 178.332
c = wt.set_machine_type('VarSpeedVarPitch')
c.vIn = 3.
c.vOut = 25.
c.minOmega = 6.
c.maxOmega = 9.6
c.minPitch = 0.
# --- 5 ---
# connect the parameterized structure vartrees to the
# structural solver
top.connect('st_builder.cs2d', 'st.cs2d')
# connect the splined planform to the blade geometry vartree
# in the aeroelastic solver
top.connect('pf_splines.pfOut', 'ae.wt.blade1.geom')
# connect computed beam structure to input vartree in the
# aeroelastic solver
top.connect('st.beam_structure', 'ae.wt.blade1.beam_structure')
# --- 6 ---
# run it
top.run()
# --- 7 ---
class ArrayTest(unittest.TestCase):
def setUp(self):
self.top = Assembly()
self.top.add('driver', MyDriver())
self.top.add('comp', ArrayComp())
self.top.driver.workflow.add('comp')
def test_basic(self):
top = self.top
comp = top.comp
driver = top.driver
comp.x1d = [1, 2, 3]
comp.x2d = [[1, 2, 3], [4, 5, 6]]
top.run()
self.assertEqual(comp.fx1d, 6.)
self.assertEqual(list(comp.fx2d), [6., 15.])
code = "driver.add_parameter('comp.x1d')"
assert_raises(self, code, globals(), locals(), ValueError,
"driver: Trying to add parameter 'comp.x1d', but no"
" lower limit was found and no 'low' argument was given."
" One or the other must be specified.")
code = "driver.add_parameter('comp.x1d', low=-10)"
assert_raises(self, code, globals(), locals(), ValueError,
"driver: Trying to add parameter 'comp.x1d', but no"
" upper limit was found and no 'high' argument was given."
" One or the other must be specified.")
driver.add_parameter('comp.x1d', low=-10, high=10, start=1)
code = "driver.add_parameter('comp.x2d')"
assert_raises(self, code, globals(), locals(), ValueError,
"driver: Trying to add parameter 'comp.x2d', but no"
" lower limit was found and no 'low' argument was given."
" One or the other must be specified.")
code = "driver.add_parameter('comp.x2d', low=-10)"
assert_raises(self, code, globals(), locals(), ValueError,
"driver: Trying to add parameter 'comp.x2d', but no"
" upper limit was found and no 'high' argument was given."
" One or the other must be specified.")
driver.add_parameter('comp.x2d', low=-10, high=10, start=2)
targets = driver.list_param_targets()
self.assertEqual(targets, ['comp.x1d', 'comp.x2d'])
targets = driver.list_param_group_targets()
self.assertEqual(targets, [('comp.x1d',), ('comp.x2d',)])
self.assertEqual(list(comp.x1d), [1, 2, 3])
self.assertEqual([list(row) for row in comp.x2d[:]],
[[1, 2, 3], [4, 5, 6]])
driver.init_parameters()
self.assertEqual(list(comp.x1d), [1, 1, 1])
self.assertEqual([list(row) for row in comp.x2d[:]],
[[2, 2, 2], [2, 2, 2]])
driver.set_parameter_by_name('comp.x1d', 3)
self.assertEqual(list(comp.x1d), [3, 3, 3])
driver.set_parameter_by_name('comp.x1d', array([4, 5, 6]))
self.assertEqual(list(comp.x1d), [4, 5, 6])
driver.set_parameter_by_name('comp.x2d', 4)
self.assertEqual([list(row) for row in comp.x2d[:]],
[[4, 4, 4], [4, 4, 4]])
driver.set_parameter_by_name('comp.x2d', array([[5, 6, 7], [8, 9, 0]]))
self.assertEqual([list(row) for row in comp.x2d[:]],
[[5, 6, 7], [8, 9, 0]])
driver.set_parameters([7, 8, 9, 1, 2, 3, 4, 5, 6])
self.assertEqual(list(comp.x1d), [7, 8, 9])
self.assertEqual([list(row) for row in comp.x2d[:]],
[[1, 2, 3], [4, 5, 6]])
self.assertEqual(driver._hasparameters.get_expr_depends(),
[('driver', 'comp'), ('driver', 'comp')])
self.assertEqual(driver._hasparameters.get_referenced_compnames(),
set(['comp']))
self.assertEqual(driver._hasparameters.get_referenced_varpaths(),
set(['comp.x1d', 'comp.x2d']))
# Still have last set_parameters() values.
self.assertEqual(list(comp.x1d), [7, 8, 9])
self.assertEqual([list(row) for row in comp.x2d[:]],
[[1, 2, 3], [4, 5, 6]])
top.run()
# Now have init_parameters() values.
self.assertEqual(list(comp.x1d), [1, 1, 1])
self.assertEqual([list(row) for row in comp.x2d[:]],
[[2, 2, 2], [2, 2, 2]])
self.assertEqual(comp.fx1d, 3)
self.assertEqual(list(comp.fx2d), [6, 6])
def test_mixed_use(self):
# Connect to one element of array and use another element as parameter.
self.top.comp.x1d = [1, 2, 3]
self.top.add('exec_comp', ExecComp(exprs=['c=x+y', 'd=x-y']))
self.top.driver.workflow.add('exec_comp')
self.top.connect('exec_comp.c', 'comp.x1d[0]')
self.top.driver.add_parameter('comp.x1d[1]', low=-10, high=10, start=1)
self.top.run()
class TestCase_Residuals(unittest.TestCase):
def setUp(self):
""" Called before each test. """
self.model = Assembly()
self.model.add('c1', MyComp())
self.model.add('c2', MyComp())
self.model.add('driver', SimpleDriver())
self.model.driver.workflow = CyclicWorkflow()
self.model.driver.workflow.add(['c1', 'c2'])
def tearDown(self):
""" Called after each test. """
self.model = None
def test_column_vector(self):
self.model.c1.add('y_a', Array(iotype='out'))
self.model.c1.y_a = array([[1.0], [2.0]])
self.model.c2.add('x_a', Array(iotype='in'))
self.model.c2.x_a = array([[3.0], [6.0]])
self.model.connect('c1.y_a', 'c2.x_a')
self.model.connect('c2.y', 'c1.x')
self.model.driver.workflow.initialize_residual()
self.model.run()
dv = array([3.0, 5.0])
self.model.driver.workflow.set_independents(dv)
indep = self.model.driver.workflow.get_independents()
self.assertEqual(indep[0], 3.0)
self.assertEqual(indep[1], 5.0)
dep = self.model.driver.workflow.get_dependents()
self.assertEqual(dep[0], -2.0)
self.assertEqual(dep[1], -3.0)
def test_row_vector(self):
self.model.c1.add('y_a', Array(iotype='out'))
self.model.c1.y_a = array([[1.0, 2.0]])
self.model.c2.add('x_a', Array(iotype='in'))
self.model.c2.x_a = array([[3.0, 6.0]])
self.model.connect('c1.y_a', 'c2.x_a')
self.model.connect('c2.y', 'c1.x')
self.model.driver.workflow.initialize_residual()
self.model.run()
dv = array([3.0, 5.0])
self.model.driver.workflow.set_independents(dv)
indep = self.model.driver.workflow.get_independents()
self.assertEqual(indep[0], 3.0)
self.assertEqual(indep[1], 5.0)
dep = self.model.driver.workflow.get_dependents()
self.assertEqual(dep[0], -2.0)
self.assertEqual(dep[1], -3.0)
def test_array_1D(self):
self.model.c1.add('y_a', Array(iotype='out'))
self.model.c1.y_a = array([1.0, 2.0])
self.model.c2.add('x_a', Array(iotype='in'))
self.model.c2.x_a = array([3.0, 6.0])
self.model.connect('c1.y_a', 'c2.x_a')
self.model.connect('c2.y', 'c1.x')
self.model.driver.workflow.initialize_residual()
self.model.run()
dv = array([3.0, 5.0])
self.model.driver.workflow.set_independents(dv)
indep = self.model.driver.workflow.get_independents()
self.assertEqual(indep[0], 3.0)
self.assertEqual(indep[1], 5.0)
dep = self.model.driver.workflow.get_dependents()
self.assertEqual(dep[0], -2.0)
self.assertEqual(dep[1], -3.0)
def test_full_matrix(self):
self.model.c1.add('y_a', Array(iotype='out'))
self.model.c1.y_a = array([[1.0, 2.0], [3.0, 4.0]])
self.model.c2.add('x_a', Array(iotype='in'))
self.model.c2.x_a = array([[2.0, 5.0], [11.0, 17.0]])
self.model.connect('c1.y_a', 'c2.x_a')
self.model.connect('c2.y', 'c1.x')
self.model.driver.workflow.initialize_residual()
self.model.run()
dv = array([3.0, 5.0, -8.0, -13.0])
self.model.driver.workflow.set_independents(dv)
indep = self.model.driver.workflow.get_independents()
self.assertEqual(indep[0], 3.0)
self.assertEqual(indep[1], 5.0)
self.assertEqual(indep[2], -8.0)
self.assertEqual(indep[3], -13.0)
dep = self.model.driver.workflow.get_dependents()
self.assertEqual(dep[0], -2.0)
self.assertEqual(dep[1], -3.0)
self.assertEqual(dep[2], 11.0)
self.assertEqual(dep[3], 17.0)
def test_matrix_element(self):
# Array element to scalar.
self.model.c1.add('y_a', Array(iotype='out'))
self.model.c1.y_a = array([[1.0, 2.0], [3.0, 4.0]])
self.model.c2.x = 7.0
self.model.connect('c1.y_a[0, 0]', 'c2.x')
self.model.connect('c2.y', 'c1.x')
self.model.driver.workflow.initialize_residual()
self.model.run()
dv = array([35.0])
self.model.driver.workflow.set_independents(dv)
indep = self.model.driver.workflow.get_independents()
self.assertEqual(indep[0], 35.0)
dep = self.model.driver.workflow.get_dependents()
self.assertEqual(dep[0], -34.0)
def test_vtree(self):
self.model.c1.add('vt_out', VarTree(Tree1(), iotype='out'))
self.model.c2.add('vt_in', VarTree(Tree1(), iotype='in'))
self.model.c2.vt_in.a1 = 4.
self.model.c2.vt_in.b1 = array([7.0, 12.0])
self.model.c2.vt_in.vt1.c1 = 13.
self.model.c2.vt_in.vt1.d1 = array([-1.0, 2.0])
self.model.connect('c1.vt_out', 'c2.vt_in')
self.model.connect('c2.y', 'c1.x')
self.model.driver.workflow.initialize_residual()
indep = self.model.driver.workflow.get_independents()
self.assertEqual(indep[0], 4.0)
self.assertEqual(indep[1], 7.0)
self.assertEqual(indep[2], 12.0)
self.assertEqual(indep[3], 13.0)
self.assertEqual(indep[4], -1.0)
self.assertEqual(indep[5], 2.0)
dep = self.model.driver.workflow.get_dependents()
self.assertEqual(dep[0], -1.0)
self.assertEqual(dep[1], -4.0)
self.assertEqual(dep[2], -5.0)
self.assertEqual(dep[3], -4.0)
self.assertEqual(dep[4], 6.0)
self.assertEqual(dep[5], 9.0)
dv = array([81.0, 82.0, 83.0, 84.0, 85.0, 86.0])
self.model.driver.workflow.set_independents(dv)
indep = self.model.driver.workflow.get_independents()
self.assertEqual(indep[0], 81.0)
self.assertEqual(indep[1], 82.0)
self.assertEqual(indep[2], 83.0)
self.assertEqual(indep[3], 84.0)
self.assertEqual(indep[4], 85.0)
self.assertEqual(indep[5], 86.0)
def test_vtree_leaf(self):
self.model.c1.add('vt_out', VarTree(Tree1(), iotype='out'))
self.model.c2.add('vt_in', VarTree(Tree1(), iotype='in'))
self.model.c2.vt_in.b1 = array([7.0, 12.0])
self.model.connect('c1.vt_out.b1', 'c2.vt_in.b1')
self.model.connect('c2.y', 'c1.x')
self.model.driver.workflow.initialize_residual()
indep = self.model.driver.workflow.get_independents()
self.assertEqual(indep[0], 7.0)
self.assertEqual(indep[1], 12.0)
dep = self.model.driver.workflow.get_dependents()
self.assertEqual(dep[0], -4.0)
self.assertEqual(dep[1], -5.0)
dv = array([13.0, 15.0])
self.model.driver.workflow.set_independents(dv)
indep = self.model.driver.workflow.get_independents()
self.assertEqual(indep[0], 13.0)
self.assertEqual(indep[1], 15.0)
class ArrayTest(unittest.TestCase):
def setUp(self):
self.top = Assembly()
self.top.add('driver', MyDriver())
self.top.add('comp', ArrayComp())
self.top.driver.workflow.add('comp')
def test_basic(self):
top = self.top
comp = top.comp
driver = top.driver
comp.x1d = [1, 2, 3]
comp.x2d = [[1, 2, 3], [4, 5, 6]]
top.run()
self.assertEqual(comp.fx1d, 6.)
self.assertEqual(list(comp.fx2d), [6., 15.])
code = "driver.add_parameter('comp.x1d')"
assert_raises(self, code, globals(), locals(), ValueError,
"driver: Trying to add parameter 'comp.x1d', but no"
" lower limit was found and no 'low' argument was given."
" One or the other must be specified.")
code = "driver.add_parameter('comp.x1d', low=-10)"
assert_raises(self, code, globals(), locals(), ValueError,
"driver: Trying to add parameter 'comp.x1d', but no"
" upper limit was found and no 'high' argument was given."
" One or the other must be specified.")
driver.add_parameter('comp.x1d', low=-10, high=10, start=1)
code = "driver.add_parameter('comp.x2d')"
assert_raises(self, code, globals(), locals(), ValueError,
"driver: Trying to add parameter 'comp.x2d', but no"
" lower limit was found and no 'low' argument was given."
" One or the other must be specified.")
code = "driver.add_parameter('comp.x2d', low=-10)"
assert_raises(self, code, globals(), locals(), ValueError,
"driver: Trying to add parameter 'comp.x2d', but no"
" upper limit was found and no 'high' argument was given."
" One or the other must be specified.")
driver.add_parameter('comp.x2d', low=-10, high=10, start=2)
top.run()
comp.x1d = [1, 2, 3]
comp.x2d = [[1, 2, 3], [4, 5, 6]]
targets = driver.list_param_targets()
self.assertEqual(targets, ['comp.x1d', 'comp.x2d'])
targets = driver.list_param_group_targets()
self.assertEqual(targets, [('comp.x1d',), ('comp.x2d',)])
self.assertEqual(list(comp.x1d), [1, 2, 3])
self.assertEqual([list(row) for row in comp.x2d[:]],
[[1, 2, 3], [4, 5, 6]])
driver.init_parameters()
var = driver.workflow._system.vec['u']['comp.x1d']
self.assertEqual(list(var), [1, 1, 1])
var = driver.workflow._system.vec['u']['comp.x2d']
self.assertEqual(list(var), [2, 2, 2, 2, 2, 2])
driver.set_parameter_by_name('comp.x1d', 3)
var = driver.workflow._system.vec['u']['comp.x1d']
self.assertEqual(list(var), [3, 3, 3])
driver.set_parameter_by_name('comp.x1d', array([4, 5, 6]))
var = driver.workflow._system.vec['u']['comp.x1d']
self.assertEqual(list(var), [4, 5, 6])
driver.set_parameter_by_name('comp.x2d', 4)
var = driver.workflow._system.vec['u']['comp.x2d']
self.assertEqual(list(var), [4, 4, 4, 4, 4, 4])
driver.set_parameter_by_name('comp.x2d', array([[5, 6, 7], [8, 9, 0]]))
var = driver.workflow._system.vec['u']['comp.x2d']
self.assertEqual(list(var), [5, 6, 7, 8, 9, 0])
driver.set_parameters([7, 8, 9, 1, 2, 3, 4, 5, 6])
var = driver.workflow._system.vec['u']['comp.x1d']
self.assertEqual(list(var), [7, 8, 9])
var = driver.workflow._system.vec['u']['comp.x2d']
self.assertEqual(list(var), [1, 2, 3, 4, 5, 6])
self.assertEqual(driver._hasparameters.get_expr_depends(),
[('driver', 'comp')])
self.assertEqual(driver._hasparameters.get_referenced_compnames(),
set(['comp']))
self.assertEqual(driver._hasparameters.get_referenced_varpaths(),
set(['comp.x1d', 'comp.x2d']))
# Still have last set_parameters() values.
var = driver.workflow._system.vec['u']['comp.x1d']
self.assertEqual(list(var), [7, 8, 9])
var = driver.workflow._system.vec['u']['comp.x2d']
self.assertEqual(list(var), [1, 2, 3, 4, 5, 6])
top.run()
# Now have init_parameters() values.
var = driver.workflow._system.vec['u']['comp.x1d']
self.assertEqual(list(var), [1, 1, 1])
var = driver.workflow._system.vec['u']['comp.x2d']
self.assertEqual(list(var), [2, 2, 2, 2, 2, 2])
def test_mixed_use(self):
# Connect to one element of array and use another element as parameter.
self.top.comp.x1d = [1., 2., 3.]
self.top.add('exec_comp', ExecComp(exprs=['c=x+y', 'd=x-y']))
self.top.driver.workflow.add('exec_comp')
self.top.connect('exec_comp.c', 'comp.x1d[0]')
self.top.driver.add_parameter('comp.x1d[1]', low=-10, high=10, start=1.)
self.top.run()
def test_set_boundary_params(self):
self.top = set_as_top(Assembly())
self.top.add('driver', MyDriver())
class ArrayComp2(Component):
x = Array([0.0, 0.0], iotype='in')
y = Array([0.0, 0.0], iotype='in')
c = Array([0.0, 0.0], iotype='out')
def execute(self):
self.c = self.x + self.y
self.top.add('comp', ArrayComp2())
self.top.driver.workflow.add('comp')
self.top.create_passthrough('comp.x')
self.top.create_passthrough('comp.c')
self.top.connect('x', 'comp.y')
self.top.driver.add_parameter(('x'), low=0., high=1e99)
self.top.x = [22.0, 31.1]
self.top.run()
self.assertEqual(self.top.x[0], 22.)
self.assertEqual(self.top.x[1], 31.1)
self.assertEqual(self.top.comp.y[0], 22.)
self.assertEqual(self.top.comp.y[1], 31.1)
class ArrayTest(unittest.TestCase):
def setUp(self):
self.top = Assembly()
self.top.add('driver', MyDriver())
self.top.add('comp', ArrayComp())
self.top.driver.workflow.add('comp')
def test_basic(self):
top = self.top
comp = top.comp
driver = top.driver
comp.x1d = [1, 2, 3]
comp.x2d = [[1, 2, 3], [4, 5, 6]]
top.run()
self.assertEqual(comp.fx1d, 6.)
self.assertEqual(list(comp.fx2d), [6., 15.])
code = "driver.add_parameter('comp.x1d')"
assert_raises(
self, code, globals(), locals(), ValueError,
"driver: Trying to add parameter 'comp.x1d', but no"
" lower limit was found and no 'low' argument was given."
" One or the other must be specified.")
code = "driver.add_parameter('comp.x1d', low=-10)"
assert_raises(
self, code, globals(), locals(), ValueError,
"driver: Trying to add parameter 'comp.x1d', but no"
" upper limit was found and no 'high' argument was given."
" One or the other must be specified.")
driver.add_parameter('comp.x1d', low=-10, high=10, start=1)
code = "driver.add_parameter('comp.x2d')"
assert_raises(
self, code, globals(), locals(), ValueError,
"driver: Trying to add parameter 'comp.x2d', but no"
" lower limit was found and no 'low' argument was given."
" One or the other must be specified.")
code = "driver.add_parameter('comp.x2d', low=-10)"
assert_raises(
self, code, globals(), locals(), ValueError,
"driver: Trying to add parameter 'comp.x2d', but no"
" upper limit was found and no 'high' argument was given."
" One or the other must be specified.")
driver.add_parameter('comp.x2d', low=-10, high=10, start=2)
targets = driver.list_param_targets()
self.assertEqual(targets, ['comp.x1d', 'comp.x2d'])
targets = driver.list_param_group_targets()
self.assertEqual(targets, [('comp.x1d', ), ('comp.x2d', )])
self.assertEqual(list(comp.x1d), [1, 2, 3])
self.assertEqual([list(row) for row in comp.x2d[:]],
[[1, 2, 3], [4, 5, 6]])
driver.init_parameters()
self.assertEqual(list(comp.x1d), [1, 1, 1])
self.assertEqual([list(row) for row in comp.x2d[:]],
[[2, 2, 2], [2, 2, 2]])
driver.set_parameter_by_name('comp.x1d', 3)
self.assertEqual(list(comp.x1d), [3, 3, 3])
driver.set_parameter_by_name('comp.x1d', array([4, 5, 6]))
self.assertEqual(list(comp.x1d), [4, 5, 6])
driver.set_parameter_by_name('comp.x2d', 4)
self.assertEqual([list(row) for row in comp.x2d[:]],
[[4, 4, 4], [4, 4, 4]])
driver.set_parameter_by_name('comp.x2d', array([[5, 6, 7], [8, 9, 0]]))
self.assertEqual([list(row) for row in comp.x2d[:]],
[[5, 6, 7], [8, 9, 0]])
driver.set_parameters([7, 8, 9, 1, 2, 3, 4, 5, 6])
self.assertEqual(list(comp.x1d), [7, 8, 9])
self.assertEqual([list(row) for row in comp.x2d[:]],
[[1, 2, 3], [4, 5, 6]])
self.assertEqual(driver._hasparameters.get_expr_depends(),
[('driver', 'comp'), ('driver', 'comp')])
self.assertEqual(driver._hasparameters.get_referenced_compnames(),
set(['comp']))
self.assertEqual(driver._hasparameters.get_referenced_varpaths(),
set(['comp.x1d', 'comp.x2d']))
# Still have last set_parameters() values.
self.assertEqual(list(comp.x1d), [7, 8, 9])
self.assertEqual([list(row) for row in comp.x2d[:]],
[[1, 2, 3], [4, 5, 6]])
top.run()
# Now have init_parameters() values.
self.assertEqual(list(comp.x1d), [1, 1, 1])
self.assertEqual([list(row) for row in comp.x2d[:]],
[[2, 2, 2], [2, 2, 2]])
self.assertEqual(comp.fx1d, 3)
self.assertEqual(list(comp.fx2d), [6, 6])
def test_mixed_use(self):
# Connect to one element of array and use another element as parameter.
self.top.comp.x1d = [1, 2, 3]
self.top.add('exec_comp', ExecComp(exprs=['c=x+y', 'd=x-y']))
self.top.driver.workflow.add('exec_comp')
self.top.connect('exec_comp.c', 'comp.x1d[0]')
self.top.driver.add_parameter('comp.x1d[1]', low=-10, high=10, start=1)
self.top.run()
class TestCase_Residuals(unittest.TestCase):
""" Test run/step/stop aspects of a simple workflow. """
def setUp(self):
""" Called before each test. """
self.model = Assembly()
self.model.add('c1', MyComp())
self.model.add('c2', MyComp())
self.model.driver.workflow = CyclicWorkflow()
self.model.driver.workflow.add(['c1', 'c2'])
def tearDown(self):
""" Called after each test. """
self.model = None
def test_column_vector(self):
self.model.c1.add('y_a', Array(iotype='out'))
self.model.c1.y_a = array([[1.0], [2.0]])
self.model.c2.add('x_a', Array(iotype='in'))
self.model.c2.x_a = array([[3.0], [6.0]])
self.model.connect('c1.y_a', 'c2.x_a')
self.model.connect('c2.y', 'c1.x')
self.model.driver.workflow.initialize_residual()
res = self.model.driver.workflow.calculate_residuals()
expected = array([[-2.0], [-4.0], [0]])
# Note, running zeros the residuals on sliced edges
self.model.run()
for j in range(len(expected)):
self.assertEqual(res[j], expected[j])
dv = array([3.0, 5.0, 0.0])
self.model.driver.workflow.set_new_state(dv)
res = self.model.driver.workflow.calculate_residuals()
expected = array([[-3.0], [-5.0], [0.0]])
for j in range(len(expected)):
self.assertEqual(res[j], expected[j])
def test_row_vector(self):
self.model.c1.add('y_a', Array(iotype='out'))
self.model.c1.y_a = array([[1.0, 2.0]])
self.model.c2.add('x_a', Array(iotype='in'))
self.model.c2.x_a = array([[3.0, 6.0]])
self.model.connect('c1.y_a', 'c2.x_a')
self.model.connect('c2.y', 'c1.x')
self.model.driver.workflow.initialize_residual()
res = self.model.driver.workflow.calculate_residuals()
expected = array([[-2.0], [-4.0], [0]])
# Note, running zeros the residuals on sliced edges
self.model.run()
for j in range(len(expected)):
self.assertEqual(res[j], expected[j])
dv = array([3.0, 5.0, 0.0])
self.model.driver.workflow.set_new_state(dv)
res = self.model.driver.workflow.calculate_residuals()
print res
expected = array([[-3.0], [-5.0], [0.0]])
for j in range(len(expected)):
self.assertEqual(res[j], expected[j])
def test_array_1D(self):
self.model.c1.add('y_a', Array(iotype='out'))
self.model.c1.y_a = array([1.0, 2.0])
self.model.c2.add('x_a', Array(iotype='in'))
self.model.c2.x_a = array([3.0, 6.0])
self.model.connect('c1.y_a', 'c2.x_a')
self.model.connect('c2.y', 'c1.x')
self.model.driver.workflow.initialize_residual()
res = self.model.driver.workflow.calculate_residuals()
expected = array([[-2.0], [-4.0], [0]])
# Note, running zeros the residuals on sliced edges
self.model.run()
for j in range(len(expected)):
self.assertEqual(res[j], expected[j])
dv = array([3.0, 5.0, 0.0])
self.model.driver.workflow.set_new_state(dv)
res = self.model.driver.workflow.calculate_residuals()
print res
expected = array([[-3.0], [-5.0], [0.0]])
for j in range(len(expected)):
self.assertEqual(res[j], expected[j])
def test_full_matrix(self):
self.model.c1.add('y_a', Array(iotype='out'))
self.model.c1.y_a = array([[1.0, 2.0], [3.0, 4.0]])
self.model.c2.add('x_a', Array(iotype='in'))
self.model.c2.x_a = array([[2.0, 5.0], [11.0, 17.0]])
self.model.connect('c1.y_a', 'c2.x_a')
self.model.connect('c2.y', 'c1.x')
self.model.driver.workflow.initialize_residual()
res = self.model.driver.workflow.calculate_residuals()
expected = array([[-1.0], [-3.0], [-8.0], [-13.0], [0]])
# Note, running zeros the residuals on sliced edges
self.model.run()
for j in range(len(expected)):
self.assertEqual(res[j], expected[j])
dv = array([3.0, 5.0, 11.0, 23.0, 0.0])
self.model.driver.workflow.set_new_state(dv)
res = self.model.driver.workflow.calculate_residuals()
print res
expected = array([[-3.0], [-5.0], [-11.0], [-23.0], [0.0]])
for j in range(len(expected)):
self.assertEqual(res[j], expected[j])
def test_matrix_element(self):
# Array element to scalar.
self.model.c1.add('y_a', Array(iotype='out'))
self.model.c1.y_a = array([[1.0, 2.0], [3.0, 4.0]])
self.model.c2.x = 7.0
self.model.connect('c1.y_a[0, 0]', 'c2.x')
self.model.connect('c2.y', 'c1.x')
self.model.driver.workflow.initialize_residual()
res = self.model.driver.workflow.calculate_residuals()
print 'matrix element res', res
expected = array([[-6.0], [0]])
# Note, running zeros the residuals on sliced edges
self.model.run()
for j in range(len(expected)):
self.assertEqual(res[j], expected[j])
dv = array([3.0, 0.0])
self.model.driver.workflow.set_new_state(dv)
res = self.model.driver.workflow.calculate_residuals()
print res
expected = array([[-3.0], [0.0]])
for j in range(len(expected)):
self.assertEqual(res[j], expected[j])
def test_vtree(self):
self.model.c1.add('vt_out', Tree1(iotype='out'))
self.model.c2.add('vt_in', Tree1(iotype='in'))
self.model.c2.vt_in.a1 = 4.
self.model.c2.vt_in.b1 = array([7.0, 12.0])
self.model.c2.vt_in.vt1.c1 = 13.
self.model.c2.vt_in.vt1.d1 = array([-1.0, 2.0])
self.model.connect('c1.vt_out', 'c2.vt_in')
self.model.connect('c2.y', 'c1.x')
self.model.driver.workflow.initialize_residual()
res = self.model.driver.workflow.calculate_residuals()
expected = array([[-1.0], [-4.0], [-5.0],
[-4.0], [6.0], [9.0],
[0]])
# Note, running zeros the residuals on sliced edges
self.model.run()
for j in range(len(expected)):
self.assertEqual(res[j], expected[j])
dv = array([3.0, 5.0, 11.0, 23.0, 12.0, 4.4, 0.0])
self.model.driver.workflow.set_new_state(dv)
res = self.model.driver.workflow.calculate_residuals()
print res
expected = array([[-3.0], [-5.0], [-11.0],
[-23.0], [-12.0], [-4.4],
[0.0]])
for j in range(len(expected)):
self.assertEqual(res[j], expected[j])
def test_vtree_leaf(self):
self.model.c1.add('vt_out', Tree1(iotype='out'))
self.model.c2.add('vt_in', Tree1(iotype='in'))
self.model.c2.vt_in.b1 = array([7.0, 12.0])
self.model.connect('c1.vt_out.b1', 'c2.vt_in.b1')
self.model.connect('c2.y', 'c1.x')
self.model.driver.workflow.initialize_residual()
res = self.model.driver.workflow.calculate_residuals()
expected = array([[-4.0], [-5.0], [0]])
# Note, running zeros the residuals on sliced edges
self.model.run()
for j in range(len(expected)):
self.assertEqual(res[j], expected[j])
dv = array([3.0, 5.0, 0.0])
self.model.driver.workflow.set_new_state(dv)
res = self.model.driver.workflow.calculate_residuals()
print res
expected = array([[-3.0], [-5.0], [0.0]])
for j in range(len(expected)):
self.assertEqual(res[j], expected[j])
class ArrayTest(unittest.TestCase):
def setUp(self):
self.top = Assembly()
self.top.add('driver', MyDriver())
self.top.add('comp', ArrayComp())
self.top.driver.workflow.add('comp')
def test_basic(self):
top = self.top
comp = top.comp
driver = top.driver
comp.x1d = [1, 2, 3]
comp.x2d = [[1, 2, 3], [4, 5, 6]]
top.run()
self.assertEqual(comp.fx1d, 6.)
self.assertEqual(list(comp.fx2d), [6., 15.])
code = "driver.add_parameter('comp.x1d')"
assert_raises(
self, code, globals(), locals(), ValueError,
"driver: Trying to add parameter 'comp.x1d', but no"
" lower limit was found and no 'low' argument was given."
" One or the other must be specified.")
code = "driver.add_parameter('comp.x1d', low=-10)"
assert_raises(
self, code, globals(), locals(), ValueError,
"driver: Trying to add parameter 'comp.x1d', but no"
" upper limit was found and no 'high' argument was given."
" One or the other must be specified.")
driver.add_parameter('comp.x1d', low=-10, high=10, start=1)
code = "driver.add_parameter('comp.x2d')"
assert_raises(
self, code, globals(), locals(), ValueError,
"driver: Trying to add parameter 'comp.x2d', but no"
" lower limit was found and no 'low' argument was given."
" One or the other must be specified.")
code = "driver.add_parameter('comp.x2d', low=-10)"
assert_raises(
self, code, globals(), locals(), ValueError,
"driver: Trying to add parameter 'comp.x2d', but no"
" upper limit was found and no 'high' argument was given."
" One or the other must be specified.")
driver.add_parameter('comp.x2d', low=-10, high=10, start=2)
top.run()
comp.x1d = [1, 2, 3]
comp.x2d = [[1, 2, 3], [4, 5, 6]]
targets = driver.list_param_targets()
self.assertEqual(targets, ['comp.x1d', 'comp.x2d'])
targets = driver.list_param_group_targets()
self.assertEqual(targets, [('comp.x1d', ), ('comp.x2d', )])
self.assertEqual(list(comp.x1d), [1, 2, 3])
self.assertEqual([list(row) for row in comp.x2d[:]],
[[1, 2, 3], [4, 5, 6]])
driver.init_parameters()
var = driver.workflow._system.vec['u']['comp.x1d']
self.assertEqual(list(var), [1, 1, 1])
var = driver.workflow._system.vec['u']['comp.x2d']
self.assertEqual(list(var), [2, 2, 2, 2, 2, 2])
driver.set_parameter_by_name('comp.x1d', 3)
var = driver.workflow._system.vec['u']['comp.x1d']
self.assertEqual(list(var), [3, 3, 3])
driver.set_parameter_by_name('comp.x1d', array([4, 5, 6]))
var = driver.workflow._system.vec['u']['comp.x1d']
self.assertEqual(list(var), [4, 5, 6])
driver.set_parameter_by_name('comp.x2d', 4)
var = driver.workflow._system.vec['u']['comp.x2d']
self.assertEqual(list(var), [4, 4, 4, 4, 4, 4])
driver.set_parameter_by_name('comp.x2d', array([[5, 6, 7], [8, 9, 0]]))
var = driver.workflow._system.vec['u']['comp.x2d']
self.assertEqual(list(var), [5, 6, 7, 8, 9, 0])
driver.set_parameters([7, 8, 9, 1, 2, 3, 4, 5, 6])
var = driver.workflow._system.vec['u']['comp.x1d']
self.assertEqual(list(var), [7, 8, 9])
var = driver.workflow._system.vec['u']['comp.x2d']
self.assertEqual(list(var), [1, 2, 3, 4, 5, 6])
self.assertEqual(driver._hasparameters.get_expr_depends(),
[('driver', 'comp')])
self.assertEqual(driver._hasparameters.get_referenced_compnames(),
set(['comp']))
self.assertEqual(driver._hasparameters.get_referenced_varpaths(),
set(['comp.x1d', 'comp.x2d']))
# Still have last set_parameters() values.
var = driver.workflow._system.vec['u']['comp.x1d']
self.assertEqual(list(var), [7, 8, 9])
var = driver.workflow._system.vec['u']['comp.x2d']
self.assertEqual(list(var), [1, 2, 3, 4, 5, 6])
top.run()
# Now have init_parameters() values.
var = driver.workflow._system.vec['u']['comp.x1d']
self.assertEqual(list(var), [1, 1, 1])
var = driver.workflow._system.vec['u']['comp.x2d']
self.assertEqual(list(var), [2, 2, 2, 2, 2, 2])
def test_mixed_use(self):
# Connect to one element of array and use another element as parameter.
self.top.comp.x1d = [1., 2., 3.]
self.top.add('exec_comp', ExecComp(exprs=['c=x+y', 'd=x-y']))
self.top.driver.workflow.add('exec_comp')
self.top.connect('exec_comp.c', 'comp.x1d[0]')
self.top.driver.add_parameter('comp.x1d[1]',
low=-10,
high=10,
start=1.)
self.top.run()
def test_set_boundary_params(self):
self.top = set_as_top(Assembly())
self.top.add('driver', MyDriver())
class ArrayComp2(Component):
x = Array([0.0, 0.0], iotype='in')
y = Array([0.0, 0.0], iotype='in')
c = Array([0.0, 0.0], iotype='out')
def execute(self):
self.c = self.x + self.y
self.top.add('comp', ArrayComp2())
self.top.driver.workflow.add('comp')
self.top.create_passthrough('comp.x')
self.top.create_passthrough('comp.c')
self.top.connect('x', 'comp.y')
self.top.driver.add_parameter(('x'), low=0., high=1e99)
self.top.x = [22.0, 31.1]
self.top.run()
self.assertEqual(self.top.x[0], 22.)
self.assertEqual(self.top.x[1], 31.1)
self.assertEqual(self.top.comp.y[0], 22.)
self.assertEqual(self.top.comp.y[1], 31.1)
class TestLazyComponent(unittest.TestCase):
def setUp(self):
self.top = Assembly()
self.top.add("t", TestComp())
self.top.add("s", SinkComp())
self.top.driver.workflow.add(["t", "s"])
self.top.set('t.a', 1)
def tearDown(self):
self.top = None
def test_full_connected(self):
self.top.connect('t.x', 's.i1')
self.top.connect('t.y', 's.i2')
self.top.connect('t.z', 's.i3')
self.top.run()
self.assertEqual(self.top.t.x, 2)
self.assertEqual(self.top.t.y, 3)
self.assertEqual(self.top.t.z, 4)
self.assertEqual(self.top.get_valid(['t.x','t.y','t.z']),
[True,True,True])
def test_partial_connected(self):
self.top.connect('t.x', 's.i1')
self.top.connect('t.y', 's.i2')
self.top.run()
self.assertEqual(self.top.t.x, 2)
self.assertEqual(self.top.t.y, 3)
self.assertEqual(self.top.t.z, 0)
self.assertEqual(self.top.get_valid(['t.x','t.y','t.z']),
[True,True,False])
#now try re-running with a different configuration to test the validy reseting
self.top.disconnect('t')
self.top.connect('t.x', 's.i1')
self.top.set('t.a', 2)
self.top.run()
self.assertEqual(self.top.t.x, 3)
self.assertEqual(self.top.t.y, 3) #this value is carried over from the first run call, but it's wrong... so not valid
self.assertEqual(self.top.t.z, 0)
self.assertEqual(self.top.get_valid(['t.x','t.y','t.z']),
[True,False,False])
#not needed right now, since we're not checking for this
#on regular component.
"""def test_partial_connect_exec_error(self):
self.top.add('t', BrokenTestComp())
self.top.connect('t.x', 's.i1')
self.top.set('t.a', 1)
self.top.run()
self.assertEqual(self.top.t.x, 10)
self.assertEqual(self.top.t.y, 0)
self.assertEqual(self.top.t.z, 0)
self.top.connect('t.y', 's.i2')
self.top.set('t.a', 2)
try:
self.top.run()
except RuntimeError as err:
msg = str(err)
self.assertEqual(msg, "t: output 'y' is connected to something in your model, but was not calculated during execution")
else:
self.fail("RuntimeError Expected")
"""
def test_new_connection_invalidation(self):
self.top.connect('t.x', 's.i1')
self.top.set('t.a', 1)
self.top.run()
self.assertEqual(self.top.t.x, 2)
self.assertEqual(self.top.t.y, 0)
self.assertEqual(self.top.t.z, 0)
self.assertEqual(self.top.get_valid(['t.x','t.y','t.z']),
[True,False,False])
#new connection is made, but no inputs are invalid. Still need to run!
self.top.connect('t.y', 's.i2')
self.top.run()
self.assertEqual(self.top.t.x, 2)
self.assertEqual(self.top.t.y, 3)
self.assertEqual(self.top.t.z, 0)
self.assertEqual(self.top.get_valid(['t.x','t.y','t.z']),
[True,True,False])
def test_dynamic_trait(self):
self.top.connect('t.x', 's.i1')
self.top.connect('t.y', 's.i2')
self.top.t.add('w', Float(0.0, iotype="out"))
self.top.connect('t.w', 's.i3')
#not checking for this yet
#try:
# self.top.run()
#except RuntimeError as err:
# msg = str(err)
# self.assertEqual(msg, "t (1-1): output 'w' is connected to something in your model, but was not calculated during execution")
#else:
# self.fail("RuntimeError Expected")
self.top.run()
self.assertEqual(self.top.get_valid(['t.w','t.x','t.y','t.z']),
[True,True,True,False])
def test_output_stays_at_default(self):
# check that validity is managed properly if outputs are calcualted,
# but their values stay at the initial/default setting
#note: this is not really necessary, unless we start testing
#that outputs were actually calculated
self.top.connect('t.x', 's.i1')
self.top.set('t.a', -1)
self.top.run()
self.assertEqual(self.top.t.x, 0)
self.assertEqual(self.top.t.y, 0)
self.assertEqual(self.top.t.z, 0)
self.assertEqual(self.top.t.get_valid(['x','y','z']),
[True,False,False])
wt.cone_angle = -2.5
wt.hub_radius = 2.8
wt.blade_length = 86.366
wt.rotor_diameter = 178.332
c = wt.set_machine_type('VarSpeedVarPitch')
c.vIn = 3.
c.vOut = 25.
c.minOmega = 6.
c.maxOmega = 9.6
c.minPitch = 0.
# --- 5 ---
# connect the parameterized structure vartrees to the
# structural solver
top.connect('st_builder.cs2d', 'st.cs2d')
# connect the splined planform to the blade geometry vartree
# in the aeroelastic solver
top.connect('pf_splines.pfOut', 'ae.wt.blade1.geom')
# connect computed beam structure to input vartree in the
# aeroelastic solver
top.connect('st.beam_structure', 'ae.wt.blade1.beam_structure')
# --- 6 ---
# run it
top.run()
# --- 7 ---
wt = top.ae.wt
wt.turbine_name = 'DTU 10MW RWT'
wt.doc = 'FUSED-Wind definition of the DTU 10MW RWT'
wt.tower_height = 115.63
wt.hub_height = 119.0
wt.towertop_height = 2.75
wt.shaft_length = 7.1
wt.tilt_angle = 5.0
wt.cone_angle = -2.5
wt.hub_radius = 2.8
wt.blade_length = 86.366
wt.rotor_diameter = 178.332
c = wt.set_machine_type('VarSpeedVarPitch')
c.vIn = 3.
c.vOut = 25.
c.minOmega = 6.
c.maxOmega = 9.6
c.minPitch = 0.
# --- 4 ---
# connect the splined planform to the blade geometry vartree
# in the aeroelastic solver
top.connect('pf_splines.pfOut', 'ae.wt.blade1.geom')
# --- 5 ---
# run it
top.run()
# --- 6 ---
wt = top.ae.wt
wt.turbine_name = 'DTU 10MW RWT'
wt.doc = 'FUSED-Wind definition of the DTU 10MW RWT'
wt.tower_height = 115.63
wt.hub_height = 119.0
wt.towertop_height = 2.75
wt.shaft_length = 7.1
wt.tilt_angle = 5.0
wt.cone_angle = -2.5
wt.hub_radius = 2.8
wt.blade_length = 86.366
wt.rotor_diameter = 178.332
c = wt.set_machine_type('VarSpeedVarPitch')
c.vIn = 3.
c.vOut = 25.
c.minOmega = 6.
c.maxOmega = 9.6
c.minPitch = 0.
# --- 4 ---
# connect the splined planform to the blade geometry vartree
# in the aeroelastic solver
top.connect('pf_splines.pfOut', 'ae.wt.blade1.geom')
# --- 5 ---
# run it
top.run()
# --- 6 ---
