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'])
# paths must be initialized prior to calling _setup_variables
group._setup_paths('')
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_variables(self):
group = Group()
group.add("C1", ExecComp("y=x*2.0"), promotes=["x"])
group.add("C2", ExecComp("y=x*2.0"), promotes=["y"])
# paths must be initialized prior to calling _setup_variables
group._setup_paths("")
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"])
root._setup_paths("")
params_dict, unknowns_dict = root._setup_variables()
# verify we get correct connection information
connections = root._get_explicit_connections()
expected_connections = {"C2.x": ["C1.y"], "C3.x": ["C1.y"]}
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'])
root._setup_paths('')
params_dict, unknowns_dict = root._setup_variables()
# verify we get correct connection information
connections = root._get_explicit_connections()
expected_connections = {'C2.x': ['C1.y'], 'C3.x': ['C1.y']}
self.assertEqual(connections, expected_connections)
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'])
# paths must be initialized prior to calling _setup_variables
group._setup_paths('')
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'])
root._setup_paths('')
params_dict, unknowns_dict = root._setup_variables()
# verify we get correct connection information
connections = root._get_explicit_connections()
expected_connections = {
'C2.x': ['C1.y'],
'C3.x': ['C1.y']
}
self.assertEqual(connections, expected_connections)
