def test_view_model_from_problem(self):
"""
Test that an n2 html file is generated from a Problem.
"""
p = Problem()
p.model = SellarStateConnection()
p.setup(check=False)
view_model(p, outfile=self.problem_filename, show_browser=False)
# Check that the html file has been created and has something in it.
self.assertTrue(os.path.isfile(self.problem_html_filename), (self.problem_html_filename + " is not a valid file."))
self.assertGreater(os.path.getsize(self.problem_html_filename), 100)
def test_view_model_from_problem(self):
"""
Test that an n2 html file is generated from a Problem.
"""
p = Problem()
p.model = SellarStateConnection()
p.setup(check=False)
view_model(p, outfile=self.problem_filename, show_browser=False)
# Check that the html file has been created and has something in it.
self.assertTrue(os.path.isfile(self.problem_html_filename), (self.problem_html_filename + " is not a valid file."))
self.assertGreater(os.path.getsize(self.problem_html_filename), 100)
def test_model_viewer_has_correct_data_from_problem(self):
"""
Verify that the correct model structure data exists when stored as compared
to the expected structure, using the SellarStateConnection model.
"""
p = Problem()
p.model = SellarStateConnection()
p.setup(check=False)
model_viewer_data = _get_viewer_data(p)
tree_json = json.dumps(model_viewer_data['tree'])
conns_json = json.dumps(model_viewer_data['connections_list'])
self.assertEqual(self.expected_tree_json, tree_json)
self.assertEqual(self.expected_conns_json, conns_json)
def test_model_viewer_has_correct_data_from_problem(self):
"""
Verify that the correct model structure data exists when stored as compared
to the expected structure, using the SellarStateConnection model.
"""
p = Problem()
p.model = SellarStateConnection()
p.setup(check=False)
model_viewer_data = _get_viewer_data(p)
tree_json = json.dumps(model_viewer_data['tree'])
conns_json = json.dumps(model_viewer_data['connections_list'])
self.assertEqual(self.expected_tree_json, tree_json)
self.assertEqual(self.expected_conns_json, conns_json)
def test_view_model_set_title(self):
"""
Test that an n2 html file is generated from a Problem.
"""
p = Problem()
p.model = SellarStateConnection()
p.setup(check=False)
view_model(p, outfile=self.problem_html_filename, show_browser=DEBUG,
title="Sellar State Connection")
# Check that the html file has been created and has something in it.
self.assertTrue(os.path.isfile(self.problem_html_filename),
(self.problem_html_filename + " is not a valid file."))
self.assertTrue( 'OpenMDAO Model Hierarchy and N2 diagram: Sellar State Connection' \
in open(self.problem_html_filename).read() )
def test_model_viewer_has_correct_data_from_problem(self):
"""
Verify that the correct model structure data exists when stored as compared
to the expected structure, using the SellarStateConnection model.
"""
p = Problem()
p.model = SellarStateConnection()
p.setup(check=False)
model_viewer_data = _get_viewer_data(p)
self.assertDictEqual(model_viewer_data['tree'], self.expected_tree)
self.assertListEqual(model_viewer_data['connections_list'],
self.expected_conns)
self.assertDictEqual(model_viewer_data['abs2prom'],
self.expected_abs2prom)
def test_view_model_from_sqlite(self):
"""
Test that an n2 html file is generated from a sqlite file.
"""
p = Problem()
p.model = SellarStateConnection()
r = SqliteRecorder(self.sqlite_db_filename2)
p.driver.add_recorder(r)
p.setup(check=False)
p.final_setup()
r.close()
view_model(self.sqlite_db_filename2, outfile=self.sqlite_filename, show_browser=False)
# Check that the html file has been created and has something in it.
self.assertTrue(os.path.isfile(self.sqlite_html_filename), (self.problem_html_filename + " is not a valid file."))
self.assertGreater(os.path.getsize(self.sqlite_html_filename), 100)
def test_view_model_from_sqlite(self):
"""
Test that an n2 html file is generated from a sqlite file.
"""
p = Problem()
p.model = SellarStateConnection()
r = SqliteRecorder(self.sqlite_db_filename2)
p.driver.add_recorder(r)
p.setup(check=False)
p.final_setup()
r.shutdown()
view_model(self.sqlite_db_filename2, outfile=self.sqlite_filename, show_browser=False)
# Check that the html file has been created and has something in it.
self.assertTrue(os.path.isfile(self.sqlite_html_filename), (self.problem_html_filename + " is not a valid file."))
self.assertGreater(os.path.getsize(self.sqlite_html_filename), 100)
def test_view_model_set_title(self):
"""
Test that an n2 html file is generated from a Problem.
"""
p = Problem()
p.model = SellarStateConnection()
p.setup()
view_model(p,
outfile=self.problem_html_filename,
show_browser=DEBUG,
title="Sellar State Connection")
# Check that the html file has been created and has something in it.
self.assertTrue(os.path.isfile(self.problem_html_filename),
(self.problem_html_filename + " is not a valid file."))
self.assertTrue( 'OpenMDAO Model Hierarchy and N2 diagram: Sellar State Connection' \
in open(self.problem_html_filename).read() )
def test_model_viewer_has_correct_data_from_sqlite(self):
"""
Verify that the correct data exists when a model structure is recorded
and then pulled out of a sqlite db file and compared to the expected
structure. Uses the SellarStateConnection model.
"""
p = Problem()
p.model = SellarStateConnection()
r = SqliteRecorder(self.sqlite_db_filename)
p.driver.add_recorder(r)
p.setup(check=False)
p.final_setup()
r.close()
model_viewer_data = _get_viewer_data(self.sqlite_db_filename)
tree_json = json.dumps(model_viewer_data['tree'])
conns_json = json.dumps(model_viewer_data['connections_list'])
self.assertEqual(self.expected_tree_json, tree_json)
self.assertEqual(self.expected_conns_json, conns_json)
def test_view_model_from_sqlite(self):
"""
Test that an n2 html file is generated from a sqlite file.
"""
p = Problem()
p.model = SellarStateConnection()
r = SqliteRecorder(self.sqlite_db_filename2)
p.driver.add_recorder(r)
p.setup(check=False)
p.final_setup()
r.shutdown()
view_model(self.sqlite_db_filename2, outfile=self.sqlite_html_filename, show_browser=DEBUG)
# Check that the html file has been created and has something in it.
self.assertTrue(os.path.isfile(self.sqlite_html_filename),
(self.problem_html_filename + " is not a valid file."))
self.assertGreater(os.path.getsize(self.sqlite_html_filename), 100)
# Check that there are no errors when running from the command line with a recording.
check_call('openmdao view_model --no_browser %s' % self.sqlite_db_filename2)
def test_model_viewer_has_correct_data_from_sqlite(self):
"""
Verify that the correct data exists when a model structure is recorded
and then pulled out of a sqlite db file and compared to the expected
structure. Uses the SellarStateConnection model.
"""
p = Problem()
p.model = SellarStateConnection()
r = SqliteRecorder(self.sqlite_db_filename)
p.driver.add_recorder(r)
p.setup(check=False)
p.final_setup()
r.close()
model_viewer_data = _get_viewer_data(self.sqlite_db_filename)
tree_json = json.dumps(model_viewer_data['tree'])
conns_json = json.dumps(model_viewer_data['connections_list'])
self.assertEqual(self.expected_tree_json, tree_json)
self.assertEqual(self.expected_conns_json, conns_json)
def test_n2_from_sqlite(self):
"""
Test that an n2 html file is generated from a sqlite file.
"""
p = Problem()
p.model = SellarStateConnection()
r = SqliteRecorder(self.sqlite_db_filename2)
p.driver.add_recorder(r)
p.setup()
p.final_setup()
r.shutdown()
n2(self.sqlite_db_filename2,
outfile=self.sqlite_html_filename,
show_browser=DEBUG)
# Check that the html file has been created and has something in it.
self.assertTrue(os.path.isfile(self.sqlite_html_filename),
(self.problem_html_filename + " is not a valid file."))
self.assertGreater(os.path.getsize(self.sqlite_html_filename), 100)
# Check that there are no errors when running from the command line with a recording.
check_call('openmdao n2 --no_browser %s' % self.sqlite_db_filename2)
self.connect('ambient.Ts', 'balance.rhs:Tt')
self.connect('balance.Pt', 'fs.P')
self.connect('balance.Tt', 'fs.T')
self.connect('Fl_O:stat:P', 'balance.lhs:Pt')
self.connect('Fl_O:stat:T', 'balance.lhs:Tt')
# self.set_order(['ambient', 'subgroup'])
if __name__ == "__main__":
from openmdao.core.problem import Problem, IndepVarComp
p1 = Problem()
p1.model = Group()
des_vars = p1.model.add_subsystem('des_vars', IndepVarComp())
des_vars.add_output('W', 0.0, units='lbm/s')
des_vars.add_output('alt', 1., units='ft')
des_vars.add_output('MN', 0.5)
des_vars.add_output('dTs', 0.0, units='degR')
fc = p1.model.add_subsystem("fc", FlightConditions())
p1.model.connect('des_vars.W', 'fc.W')
p1.model.connect('des_vars.alt', 'fc.alt')
p1.model.connect('des_vars.MN', 'fc.MN')
p1.model.connect('des_vars.dTs', 'fc.dTs')
p1.setup()
])
if __name__ == "__main__":
from openmdao.core.problem import Problem
from pycycle.elements.api import FlowStart
from pycycle.cea import species_data
from pycycle.constants import AIR_MIX, AIR_FUEL_MIX
from pycycle.connect_flow import connect_flow
from openmdao.api import IndepVarComp
from pycycle.maps.lpt2269 import LPT2269
gas = AIR_FUEL_MIX
prob = Problem()
prob.model = Group()
dv = prob.model.add_subsystem('des_vars', IndepVarComp(), promotes=['*'])
dv.add_output('P', 15.8172, units='lbf/inch**2')
dv.add_output('T', 2644.02, units='degR')
dv.add_output('W', 100.0, units='lbm/s')
dv.add_output('eff', 0.9, units=None)
dv.add_output('P_bld', 16.000, units='lbf/inch**2')
dv.add_output('T_bld', 2000.0, units='degR')
dv.add_output('W_bld', 1.0, units='lbm/s')
dv.add_output('PR', 4.0, units=None)
dv.add_output('frac_P', 0.5),
prob.model.add_subsystem(
'flow_start',
FlowStart(thermo_data=species_data.janaf, elements=AIR_MIX))