def testCreateParameterDict(self):
""" fit.create_parameter_dict() """
pars = {'logg_value': 4.0, 'vrad_vary': False, 'z_value': -0.3, 'vrad_value': 0,
'logg_vary': True, 'rv_min': 2.1, 'vrad_min': 0, 'vrad_max': 0, 'z_max': 0.0,
'ebv_max': 0.015, 'teff_value': 6000, 'z_vary': True, 'rv_value': 2.4,
'teff_vary': True, 'logg_min': 3.5, 'rv_max': 3.1, 'z_min': -0.5,
'ebv_min': 0.005, 'teff_min': 5000, 'logg_max': 4.5, 'ebv_value': 0.007,
'teff_max': 7000, 'rv_vary': True, 'ebv_vary': True}
exp = {'max': array([3.1, 7000, 4.5, 0.015, 0, 0.0], dtype=object),
'vary': array([True, True, True, True, False, True], dtype=object),
'names': array(['rv', 'teff', 'logg', 'ebv', 'vrad', 'z'], dtype='|S4'),
'value': array([2.4, 6000, 4.0, 0.007, 0, -0.3], dtype=object),
'min': array([2.1, 5000, 3.5, 0.005, 0, -0.5], dtype=object)}
res = fit.create_parameter_dict(**pars)
self.assertArrayEqual(res['max'], exp['max'])
self.assertArrayEqual(res['vary'], exp['vary'])
self.assertArrayEqual(res['min'], exp['min'])
self.assertArrayEqual(res['names'], exp['names'])
self.assertArrayEqual(res['value'], exp['value'])
def testCreateParameterDict(self):
""" fit.create_parameter_dict() """
pars = dict(teff_value=5000, teff_min=4000, teff_max=6000,
logg_value=4.5, logg_min=3.5, logg_max=5.5,
ebv_value=0.001, ebv_min=0.0, ebv_max=0.01, ebv_vary=False,
z_value=0, z_vary=False,
rad_expr='G*sqrt(2*logg)/m')
parameters = fit.create_parameter_dict(**pars)
self.assertTrue('value' in parameters)
self.assertTrue('min' in parameters)
self.assertTrue('max' in parameters)
self.assertTrue('vary' in parameters)
self.assertTrue('expr' in parameters)
names = parameters['names']
self.assertTrue('teff' in names)
self.assertTrue('logg' in names)
self.assertTrue('ebv' in names)
self.assertTrue('z' in names)
self.assertTrue('rad' in names)
for key in parameters.keys():
self.assertTrue(type(parameters[key]) == np.ndarray)
self.assertEqual(parameters['value'][names == 'teff'], 5000)
self.assertEqual(parameters['value'][names == 'logg'], 4.5)
self.assertEqual(parameters['value'][names == 'ebv'], 0.001)
self.assertEqual(parameters['value'][names == 'z'], 0)
self.assertEqual(parameters['min'][names == 'teff'], 4000)
self.assertEqual(parameters['min'][names == 'logg'], 3.5)
self.assertEqual(parameters['min'][names == 'ebv'], 0.0)
self.assertEqual(parameters['max'][names == 'teff'], 6000)
self.assertEqual(parameters['max'][names == 'logg'], 5.5)
self.assertEqual(parameters['max'][names == 'ebv'], 0.01)
self.assertEqual(parameters['vary'][names == 'ebv'], False)
self.assertEqual(parameters['expr'][names == 'rad'], 'G*sqrt(2*logg)/m')
