def test_simple_array(self): prob = om.Problem() prob.model.add_subsystem('a_comp', ArrayMultiplyComp(array=np.array([1, 2, 3]))) prob.setup() prob.set_val('a_comp.x', 5.) prob.run_model() assert_near_equal(prob.get_val('a_comp.y'), [5., 10., 15.])
def test_simple_array(self): import numpy as np import openmdao.api as om from openmdao.test_suite.components.options_feature_array import ArrayMultiplyComp prob = om.Problem() prob.model.add_subsystem('a_comp', ArrayMultiplyComp(array=np.array([1, 2, 3]))) prob.setup() prob.set_val('a_comp.x', 5.) prob.run_model() assert_near_equal(prob.get_val('a_comp.y'), [5., 10., 15.])
def test_simple_array(self): import numpy as np import openmdao.api as om from openmdao.test_suite.components.options_feature_array import ArrayMultiplyComp prob = om.Problem() prob.model.add_subsystem('inputs', om.IndepVarComp('x', 1.)) prob.model.add_subsystem('a_comp', ArrayMultiplyComp(array=np.array([1, 2, 3]))) prob.model.connect('inputs.x', 'a_comp.x') prob.setup() prob['inputs.x'] = 5. prob.run_model() assert_rel_error(self, prob['a_comp.y'], [5., 10., 15.])