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)