def tick(self):
v = get_parameters().get('risk_tolerance_mean') - 0.1
if v < 0.1:
v = 0.1
logger().debug(f'Global risk_tolerance change to {v}')
change_parameters(risk_tolerance_mean=v)
self.model.reroll_human_properties()
def test_reached_hospitalization_limit():
assert not model.reached_hospitalization_limit()
base.change_parameters(hospitalization_capacity=0.5)
# Hospitalizing population
old_total_hospitalized = model.global_count.total_hospitalized
model.global_count.total_hospitalized = model.global_count.total_population / 2 - 1
assert not model.reached_hospitalization_limit()
model.global_count.total_hospitalized += 1
assert model.reached_hospitalization_limit()
model.global_count.total_hospitalized = old_total_hospitalized
def end_cycle(self, model):
if self.state == 0:
if (self.location.covid_model.global_count.infected_count /
location.size) >= self.perc1:
self.state = 1
change_parameters(symptomatic_isolation_rate=0.9,
asymptomatic_isolation_rate=0.8)
elif self.state == 1:
if (self.location.covid_model.global_count.infected_count /
location.size) <= (1.0 - self.perc2):
self.state = 2
change_parameters(symptomatic_isolation_rate=0.0,
asymptomatic_isolation_rate=0.0)
def tick(self):
v = get_parameters().get('risk_tolerance_mean') - 0.1
if v < 0.1:
v = 0.1
change_parameters(risk_tolerance_mean=v)
self.model.reroll_human_properties()
def test_human_disease_stages():
# Test infection
common_human = human.Human.factory(covid_model=model, forced_age=None)
common_human.immune = False
immune_human = copy.deepcopy(common_human)
immune_human.immune = True
common_human.infection_status = InfectionStatus.SUSCEPTIBLE
immune_human.infect()
common_human.infect()
assert not immune_human.is_infected()
assert common_human.is_infected()
# Test disease evolution
# ASYMPTOMATIC
assert common_human.disease_severity == DiseaseSeverity.ASYMPTOMATIC
common_human.infection_days_count = common_human.infection_incubation
common_human.disease_evolution()
assert common_human.disease_severity == DiseaseSeverity.LOW
assert common_human.is_symptomatic()
assert common_human.is_contagious()
# LOW
common_human.infection_days_count = common_human.infection_incubation + common_human.mild_duration
recover_human = copy.deepcopy(common_human)
common_human.moderate_severity_prob = 1.0
recover_human.moderate_severity_prob = 0.0
model.global_count.total_population = 2
common_human.disease_evolution()
recover_human.disease_evolution()
assert common_human.disease_severity == DiseaseSeverity.MODERATE
assert common_human.hospitalized
assert not recover_human.is_infected()
# MODERATE
common_human.infection_days_count = common_human.infection_incubation + common_human.mild_duration + common_human.hospitalization_duration - 1
common_human.hospitalized = True
common_human.death_mark = False
common_human.high_severity_prob = 1.0
recover_human = copy.deepcopy(
common_human) # recovered due to low "high_severity_probability"
recover_human.high_severity_prob = 0.0
recover_human.disease_evolution()
assert not recover_human.is_infected()
dead_human = copy.deepcopy(common_human) # death due to death_mark
dead_human.death_mark = True
dead_human.disease_evolution()
assert dead_human.is_dead
dead_human = copy.deepcopy(
common_human) # death due to lack of hospitalization
dead_human.hospitalized = False
dead_human.disease_evolution()
assert dead_human.is_dead
dead_human = copy.deepcopy(
common_human) # death due to ICU limit being reached
dead_human.hospitalized = True
dead_human.death_mark = False
model.global_count.high_severity_count = 1
model.global_count.total_population = 2
base.change_parameters(icu_capacity=0.5)
assert model.reached_icu_limit()
dead_human.disease_evolution()
assert dead_human.is_dead
base.change_parameters(icu_capacity=1.0)
model.global_count.high_severity_count = 0
assert not model.reached_icu_limit()
common_human.hospitalized = True
common_human.disease_evolution()
assert not common_human.is_dead
assert common_human.disease_severity == DiseaseSeverity.HIGH
# HIGH
common_human.infection_days_count = common_human.infection_incubation\
+ common_human.mild_duration\
+ common_human.hospitalization_duration\
+ common_human.icu_duration\
- 1
dead_human = copy.deepcopy(common_human)
dead_human.death_mark = True
dead_human.disease_evolution()
assert dead_human.is_dead
common_human.death_mark = False
common_human.disease_evolution()
assert not common_human.is_infected()
def test_human_is_wearing_mask():
sample_human = human.Human.factory(covid_model=model, forced_age=None)
base.change_parameters(mask_user_rate=1.0)
assert sample_human.is_wearing_mask()
base.change_parameters(mask_user_rate=0.0)
assert not sample_human.is_wearing_mask()
def test_human_is_isolated():
adult_human = human.Human.factory(covid_model=model, forced_age=40)
student_human = human.Human.factory(covid_model=model, forced_age=15)
elder_human = human.Human.factory(covid_model=model, forced_age=80)
# Symptomatic human
base.change_parameters(symptomatic_isolation_rate=1.0)
adult_human.immune = False
adult_human.infection_status = InfectionStatus.SUSCEPTIBLE
adult_human.infect()
adult_human.infection_days_count = adult_human.infection_incubation
assert adult_human.infection_status == InfectionStatus.INFECTED
assert adult_human.is_symptomatic()
assert adult_human.is_isolated()
base.change_parameters(symptomatic_isolation_rate=0.0)
assert not adult_human.is_isolated()
# Asymptomatic humans
adult_human.infection_days_count = adult_human.infection_incubation - 1
assert not adult_human.is_symptomatic()
# Adult (different policies and risk tolerance)
base.change_parameters(social_policies=[SocialPolicy.LOCKDOWN_RETAIL])
adult_human.work_info.work_class = WorkClasses.RETAIL
adult_human.properties.risk_tolerance = 1.0
adult_human.properties.herding_behavior = 0.0
assert not adult_human.is_isolated()
adult_human.properties.risk_tolerance = 0.0
adult_human.properties.herding_behavior = 0.0
assert adult_human.is_isolated()
base.change_parameters(social_policies=[])
adult_human.properties.risk_tolerance = 0.0
adult_human.properties.herding_behavior = 0.0
assert not adult_human.is_isolated()
# Student
base.change_parameters(social_policies=[SocialPolicy.LOCKDOWN_HIGH_SCHOOL])
assert student_human.is_isolated()
base.change_parameters(
social_policies=[SocialPolicy.LOCKDOWN_MIDDLE_SCHOOL])
assert not student_human.is_isolated()
# Elder
assert not elder_human.is_isolated()