Source code for openmdao.test_suite.components.options_feature_array

A component that multiplies an array by an input value, where
the array is given as an option of type 'numpy.ndarray'.
import numpy as np

import openmdao.api as om

[docs]class ArrayMultiplyComp(om.ExplicitComponent):
[docs] def initialize(self): self.options.declare('array', types=np.ndarray)
[docs] def setup(self): array = self.options['array'] self.add_input('x', 1.) self.add_output('y', shape=array.shape)
[docs] def setup_partials(self): self.declare_partials(of='*', wrt='*')
[docs] def compute(self, inputs, outputs): outputs['y'] = self.options['array'] * inputs['x']