def calc_probability_LD_migration_simple(person):
"Calculates the probability of migration for an agent."
age = person.get_age_months()
probability_index = get_probability_index(age, prob_time_units)
if person.get_sex() == 'female':
return migration_probabilities_female[probability_index]
elif person.get_sex() == 'male':
return migration_probabilities_male[probability_index]
def calc_probability_death(person):
"Calculates the probability of death for an agent."
age = person.get_age_months()
probability_index = get_probability_index(age, prob_time_units)
try:
if person.get_sex() == 'female':
return death_probabilities_female[probability_index]
elif person.get_sex() == 'male':
return death_probabilities_male[probability_index]
except IndexError:
raise IndexError("error calculating death probability (index %s)"%(probability_index))
def calc_probability_marriage_simple(person):
"""
Calculate the probability of marriage using a simple sex and age dependent
probability distribution.
"""
age = person.get_age_months()
probability_index = get_probability_index(age, prob_time_units)
if person.get_sex() == 'female':
return marriage_probabilities_female[probability_index]
elif person.get_sex() == 'male':
return marriage_probabilities_male[probability_index]
