Source code for openmdao.test_suite.components.simple_comps

""" Defines a few simple comps that are used in tests."""

import numpy as np

import openmdao.api as om


[docs]class DoubleArrayComp(om.ExplicitComponent): """ A fairly simple array component. """
[docs] def __init__(self): super(DoubleArrayComp, self).__init__() self.JJ = np.array([[1.0, 3.0, -2.0, 7.0], [6.0, 2.5, 2.0, 4.0], [-1.0, 0.0, 8.0, 1.0], [1.0, 4.0, -5.0, 6.0]]) self.declare_partials('*', '*')
[docs] def setup(self): # Params self.add_input('x1', np.zeros([2])) self.add_input('x2', np.zeros([2])) # Unknowns self.add_output('y1', np.zeros([2])) self.add_output('y2', np.zeros([2])) # Derivs self.declare_partials(of='*', wrt='*')
[docs] def compute(self, inputs, outputs): """ Execution. """ outputs['y1'] = self.JJ[0:2, 0:2].dot(inputs['x1']) + \ self.JJ[0:2, 2:4].dot(inputs['x2']) outputs['y2'] = self.JJ[2:4, 0:2].dot(inputs['x1']) + \ self.JJ[2:4, 2:4].dot(inputs['x2'])
[docs] def compute_partials(self, inputs, partials): """ Analytical derivatives. """ partials[('y1', 'x1')] = self.JJ[0:2, 0:2] partials[('y1', 'x2')] = self.JJ[0:2, 2:4] partials[('y2', 'x1')] = self.JJ[2:4, 0:2] partials[('y2', 'x2')] = self.JJ[2:4, 2:4]
[docs]class NonSquareArrayComp(om.ExplicitComponent): """ A fairly simple array component. """
[docs] def __init__(self): super(NonSquareArrayComp, self).__init__() self.JJ = np.array([[1.0, 3.0, -2.0, 7.0], [6.0, 2.5, 2.0, 4.0], [-1.0, 0.0, 8.0, 1.0], [1.0, 4.0, -5.0, 6.0]]) self.declare_partials('*', '*')
[docs] def setup(self): # Params self.add_input('x1', np.zeros([2])) self.add_input('x2', np.zeros([2])) # Unknowns self.add_output('y1', np.zeros([3])) self.add_output('y2', np.zeros([1])) # Derivs self.declare_partials(of='*', wrt='*')
[docs] def compute(self, inputs, outputs): """ Execution. """ outputs['y1'] = self.JJ[0:3, 0:2].dot(inputs['x1']) + \ self.JJ[0:3, 2:4].dot(inputs['x2']) outputs['y2'] = self.JJ[3:4, 0:2].dot(inputs['x1']) + \ self.JJ[3:4, 2:4].dot(inputs['x2'])
[docs] def compute_partials(self, inputs, partials): """ Analytical derivatives. """ partials[('y1', 'x1')] = self.JJ[0:3, 0:2] partials[('y1', 'x2')] = self.JJ[0:3, 2:4] partials[('y2', 'x1')] = self.JJ[3:4, 0:2] partials[('y2', 'x2')] = self.JJ[3:4, 2:4]
[docs]class TestExplCompDeprecated(om.ExplicitComponent): """ A component that adds variables in the __init__ function. """
[docs] def __init__(self): super(TestExplCompDeprecated, self).__init__() self.add_input('x1', np.zeros([2])) self.add_output('y1', np.zeros([2])) # Derivs self.declare_partials(of='*', wrt='*')
[docs] def compute(self, inputs, outputs): """ Execution. """ outputs['y1'] = 2.*inputs['x1']