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')