def test_summary(self):
prob = Problem(model=CircleOpt())
prob.setup()
prob.run_driver()
stdout = StringIO()
config_summary(prob, stream=stdout)
text = stdout.getvalue().split('\n')
expected = [
"============== Problem Summary ============",
"Groups: 1", "Components: 7",
"Max tree depth: 1", "",
"Design variables: 3 Total size: 21",
"Constraints (nonlinear): 4 Total size: 21",
"Constraints (linear): 1 Total size: 1",
"Objectives: 1 Total size: 1", "",
"Input variables: 11 Total size: 82",
"Output variables: 9 Total size: 67", "",
"Total connections: 10 Total transfer data size: 72", "",
"Driver type: Driver", "Linear Solvers: ['LinearRunOnce']",
"Nonlinear Solvers: ['NonlinearRunOnce']"
]
for i in range(len(expected)):
self.assertEqual(
text[i], expected[i],
'\nExpected: %s\nReceived: %s\n' % (expected[i], text[i]))
def summary(prob):
config_summary(prob)
sys.exit(0)
g = p.model
if 'gmres' in sys.argv:
from openmdao.solvers.linear.scipy_iter_solver import ScipyKrylov
g.linear_solver = ScipyKrylov()
g.add_subsystem("P", IndepVarComp('x', numpy.ones(vec_size)))
g.add_design_var("P.x")
par = g.add_subsystem("par", ParallelGroup())
for pt in range(pts):
ptname = "G%d" % pt
ptg = par.add_subsystem(ptname, SubGroup())
#create_dyncomps(ptg, num_comps, 2, 2, 2,
#var_factory=lambda: numpy.zeros(vec_size))
g.connect("P.x", "par.%s.C0.i0" % ptname)
#cname = ptname + '.' + "C%d"%(num_comps-1)
#g.add_objective("par.%s.o0" % cname)
#g.add_constraint("par.%s.o1" % cname, lower=0.0)
p.setup()
p.final_setup()
p.run_model()
#
from openmdao.devtools.memory import max_mem_usage
print("mem:", max_mem_usage())
config_summary(p)
def summary(prob):
config_summary(prob)
exit()
def summary(prob):
config_summary(prob)
exit()
g = p.model
if 'gmres' in sys.argv:
from openmdao.solvers.linear.scipy_iter_solver import ScipyKrylov
p.root.linear_solver = ScipyKrylov()
g.add_subsystem("P", IndepVarComp('x', numpy.ones(vec_size)))
g.add_design_var("P.x")
par = g.add_subsystem("par", ParallelGroup())
for pt in range(pts):
ptname = "G%d"%pt
ptg = par.add_subsystem(ptname, SubGroup())
#create_dyncomps(ptg, num_comps, 2, 2, 2,
#var_factory=lambda: numpy.zeros(vec_size))
g.connect("P.x", "par.%s.C0.i0" % ptname)
#cname = ptname + '.' + "C%d"%(num_comps-1)
#g.add_objective("par.%s.o0" % cname)
#g.add_constraint("par.%s.o1" % cname, lower=0.0)
p.setup(vector_class=vec_class)
p.final_setup()
p.run_model()
#
from openmdao.devtools.debug import max_mem_usage
print("mem:", max_mem_usage())
config_summary(p)
