def test_variables(self): group = Group() group.add('C1', ExecComp('y=x*2.0'), promotes=['x']) group.add("C2", ExecComp('y=x*2.0'), promotes=['y']) prob = Problem() # paths must be initialized prior to calling _setup_variables group._init_sys_data('', prob._probdata) params_dict, unknowns_dict = group._setup_variables() self.assertEqual(list(params_dict.keys()), ['C1.x', 'C2.x']) self.assertEqual(list(unknowns_dict.keys()), ['C1.y', 'C2.y']) self.assertEqual([m['promoted_name'] for n,m in params_dict.items()], ['x', 'C2.x']) self.assertEqual([m['promoted_name'] for n,m in unknowns_dict.items()], ['C1.y', 'y'])
def test_multiple_connect(self): root = Group() C1 = root.add("C1", ExecComp("y=x*2.0")) C2 = root.add("C2", ExecComp("y=x*2.0")) C3 = root.add("C3", ExecComp("y=x*2.0")) root.connect("C1.y", ["C2.x", "C3.x"]) prob = Problem() root._init_sys_data("", prob._probdata) params_dict, unknowns_dict = root._setup_variables() # verify we get correct connection information connections = root._get_explicit_connections() expected_connections = {"C2.x": [("C1.y", None)], "C3.x": [("C1.y", None)]} self.assertEqual(connections, expected_connections)
def test_multiple_connect(self): root = Group() C1 = root.add('C1', ExecComp('y=x*2.0')) C2 = root.add('C2', ExecComp('y=x*2.0')) C3 = root.add('C3', ExecComp('y=x*2.0')) root.connect('C1.y',['C2.x', 'C3.x']) prob = Problem() root._init_sys_data('', prob._probdata) params_dict, unknowns_dict = root._setup_variables() # verify we get correct connection information connections = root._get_explicit_connections() expected_connections = { 'C2.x': [('C1.y', None)], 'C3.x': [('C1.y', None)] } self.assertEqual(connections, expected_connections)
def test_multiple_connect(self): root = Group() C1 = root.add('C1', ExecComp('y=x*2.0')) C2 = root.add('C2', ExecComp('y=x*2.0')) C3 = root.add('C3', ExecComp('y=x*2.0')) root.connect('C1.y', ['C2.x', 'C3.x']) prob = Problem() root._init_sys_data('', prob._probdata) params_dict, unknowns_dict = root._setup_variables() # verify we get correct connection information connections = root._get_explicit_connections() expected_connections = { 'C2.x': [('C1.y', None)], 'C3.x': [('C1.y', None)] } self.assertEqual(connections, expected_connections)