def setUp(self): unittest.TestCase.setUp(self) self.MP = MatrixPool() epsAxis = Axis({ 'name': 'eps', 'valueL': [10., 20., 30., 40., 50.], 'units': '', 'transform': 'log10' }) pcAxis = Axis({ 'name': 'pc', 'valueL': [100., 200., 300, 400], 'units': 'psia', 'transform': 'log10' }) mrAxis = Axis({ 'name': 'mr', 'valueL': [1, 2, 3], 'units': '', 'transform': '' }) for A in [epsAxis, pcAxis, mrAxis]: self.MP.add_axis(A) M = self.MP.add_matrix(name='cea_isp', units='sec', axisNameL=['eps', 'pc', 'mr']) for eps in epsAxis: for pc in pcAxis: for mr in mrAxis: M.setByName(pc=pc, eps=eps, mr=mr, val=eps + pc + mr)
from m_pool.matrix_pool import MatrixPool from m_pool.axis_obj import Axis import itertools MP = MatrixPool(name='N2O4_MMH') epsAxis = Axis({ 'name': 'eps', 'valueL': [10., 20., 30., 40., 50.], 'units': '', 'transform': 'log10' }) pcAxis = Axis({ 'name': 'pc', 'valueL': [100., 200., 300, 400], 'units': 'psia', 'transform': 'log10' }) mrAxis = Axis({ 'name': 'mr', 'valueL': [1, 2, 3], 'units': '', 'transform': '' }) for A in [epsAxis, pcAxis, mrAxis]: MP.add_axis(A) M = MP.add_matrix(name='cea_isp', units='sec', axisNameL=['eps', 'pc', 'mr']) for eps in epsAxis: for pc in pcAxis: for mr in mrAxis: M.setByName(pc=pc, eps=eps, mr=mr, val=eps + pc + mr)
from m_pool.matrix_pool import MatrixPool from m_pool.axis_obj import Axis MP = MatrixPool(name='CHECK') epsAxis = Axis({ 'name': 'eps', 'valueL': [10., 20., 30., 40., 50.], 'units': '', 'transform': '' }) pcAxis = Axis({ 'name': 'pc', 'valueL': [100., 200., 300, 400], 'units': 'psia', 'transform': '' }) mrAxis = Axis({ 'name': 'mr', 'valueL': [1, 2, 3, 4], 'units': '', 'transform': '' }) for A in [epsAxis, pcAxis, mrAxis]: MP.add_axis(A) def chkfunc(pc, eps, mr): #return eps + 2.0*pc + 0.1*mr #return -(((eps-25.0)**2) + 2.0*((pc-225.)**2) + 333.333*(mr-1.75)**2) return (62267.0 - (((eps - 25.0)**2) + 2.0 * ((pc - 225.)**2) + 333.333 * (mr - 1.75)**2)) * (300.0 / 62267.0) + 30.0