def test_equals():
param1 = parameters.Parameters()
param2 = parameters.Parameters()
param3 = parameters.Parameters()
param4 = parameters.Parameters()
assert param1 == param2
param3.teff = 10000.
assert param1 != param3
param4.chemical_composition["H"] = 1
assert param1 != param4
def test_default_parameters():
param = parameters.Parameters()
assert param.teff == 5777
assert param.logg == 4.44
assert param.metallicity == 0.0
assert param.microturbulence == 2.0
def test_chemical_composition():
param = parameters.Parameters()
assert param.chemical_composition["H"] == 0.9204
assert param.chemical_composition["He"] == 0.07834
assert param.chemical_composition["C"] == -3.52
assert param.chemical_composition["N"] == -4.12
assert param.chemical_composition["O"] == -3.21
assert param.chemical_composition["Si"] == -4.49
def test_change_chemical_composition():
param = parameters.Parameters()
param.chemical_composition["C"] = -3.40
assert param.chemical_composition["C"] == -3.40
assert param.chemical_composition[6] == -3.40
param.chemical_composition[7] = -3.8
assert param.chemical_composition["N"] == -3.8
assert param.chemical_composition[7] == -3.8
def test_set_paramters():
param = parameters.Parameters()
param.teff = 3000.
assert param.teff == 3000.
param.logg = 4.0
assert param.logg == 4.0
param.metallicity = 1.0
assert param.metallicity == 1.0
param.microturbulence = 3.
assert param.microturbulence == 3.0
def test_update_composition_from_dict():
param = parameters.Parameters()
new_composition = {
3: -11.92,
4: -13.64,
5: -9.10,
6: -4.52,
7: -3.12,
8: -3.21
}
param.update_chemical_composition(new_composition)
assert np.alltrue([
new_composition[k] == param.chemical_composition[k]
for k in new_composition
])
def test_iterate_composition():
param = parameters.Parameters()
x = {k: param.chemical_composition[k] for k in param.chemical_composition}
assert x[1] == 0.9204
assert x[100] == None