def setup_test_binary_star():
star1 = setup_test_star()
star2 = setup_test_star()
binary = combined_light_predictor.BinaryStar()
binary.star1 = star1
binary.star2 = star2
binary.distance_modulus = star1.distance_modulus
return binary
def scenario1(dm):
"""Define a main sequence binary consisting of a K-dwarf and an M-dwarf"""
kstar = Kdwarf(name='K star', dm=dm)
mstar = Mdwarf(name='M star', dm=dm)
binary = combined_light_predictor.BinaryStar()
binary.distance_modulus = dm
binary.star1 = kstar
binary.star2 = mstar
binary.calculate_combined_light()
binary.calculate_apparent_magnitudes()
binary.calculate_combined_colours()
return binary
def scenario3(dm=None):
"""Define a binary consisting of a mid-range temperature white dwarf
plus a main sequence M-star"""
mstar = Mdwarf(name='M star', dm=dm)
wd = WhiteDwarf(20000.0,dm=dm)
binary = combined_light_predictor.BinaryStar()
binary.distance_modulus = dm
binary.star1 = wd
binary.star2 = mstar
binary.calculate_combined_light()
binary.calculate_apparent_magnitudes()
binary.calculate_combined_colours()
return binary
def scenario1(dm, extinction):
"""Define a main sequence binary consisting of a K-dwarf and an M-dwarf"""
m3star = M3dwarf(name='K star', dm=dm)
m7star = M7dwarf(name='M star', dm=dm)
binary = combined_light_predictor.BinaryStar()
binary.distance_modulus = dm
binary.star1 = m3star
binary.star2 = m7star
binary.calculate_combined_light()
binary.calculate_apparent_magnitudes()
binary.calculate_combined_colours()
binary.apply_extinction(extinction)
return binary
def scenario2(dm, extinction):
"""Define a binary consisting of a mid-range temperature white dwarf
plus a main sequence K-star"""
kstar = Kdwarf(name='K star', dm=dm)
wd = WhiteDwarf(20000.0, dm=dm)
binary = combined_light_predictor.BinaryStar()
binary.distance_modulus = dm
binary.star1 = kstar
binary.star2 = wd
binary.calculate_combined_light()
binary.calculate_apparent_magnitudes()
binary.calculate_combined_colours()
binary.apply_extinction(extinction)
return binary