def test_sc0_base_lockdown(self):
params = get_default_params()
params[nindividual_key] = 100
params[nday_key] = 30
env_dic = get_environment_simulation(params)
stats_result = sc0_base_lockdown.launch_run(params, env_dic, display_progress=False)
self.assertTrue(len(stats_result) > 0)
def test_sc6_travelers(self):
params = get_default_params()
params[nindividual_key] = 100
params[nday_key] = 30
params[additional_scenario_params_key] = [5]
env_dic = get_environment_simulation(params)
stats_result = sc6_travelers.launch_run(params, env_dic, display_progress=False)
self.assertTrue(len(stats_result) > 0)
def test_sc2_yoyo_lockdown_removal(self):
params = get_default_params()
params[nindividual_key] = 100
params[nday_key] = 30
params[additional_scenario_params_key] = [14, 2, 7]
env_dic = get_environment_simulation(params)
stats_result = sc2_yoyo_lockdown_removal.launch_run(params, env_dic, display_progress=False)
self.assertTrue(len(stats_result) > 0)
def test_sc0_base_lockdown(self):
params = get_default_params()
params[nindividual_key] = 100
params[nday_key] = 30
env_dic = get_environment_simulation(params)
ray_params = ray.put(params)
ray_env_dic = ray.put(env_dic)
pb = ProgressBar(params[nday_key])
actor = pb.actor
stats_l = [
sc0_base_lockdown.do_parallel_run.remote(ray_env_dic, ray_params,
0, 0, actor)
]
self.assertTrue(len(stats_l) > 0)
def test_sc2_yoyo_lockdown_removal(self):
params = get_default_params()
params[nindividual_key] = 100
params[nday_key] = 30
params[additional_scenario_params_key] = [14, 2, 7]
env_dic = get_environment_simulation(params)
ray_params = ray.put(params)
ray_env_dic = ray.put(env_dic)
pb = ProgressBar(params[nday_key])
actor = pb.actor
stats_l = [
sc2_yoyo_lockdown_removal.do_parallel_run.remote(
ray_env_dic, ray_params, 0, 0, actor)
]
self.assertTrue(len(stats_l) > 0)
def test_sc5_rogue_neighborhood(self):
params = get_default_params()
params[nindividual_key] = 100
params[nday_key] = 30
params[additional_scenario_params_key] = [4, 2]
env_dic = get_environment_simulation(params)
ray_params = ray.put(params)
ray_env_dic = ray.put(env_dic)
pb = ProgressBar(params[nday_key])
actor = pb.actor
stats_l = [
sc5_rogue_neighborhood.do_parallel_run.remote(
ray_env_dic, ray_params, 0, 0, actor)
]
self.assertTrue(len(stats_l) > 0)
import joblib
import numpy
from scenario.example import sc1_simple_lockdown_removal, sc2_yoyo_lockdown_removal, sc0_base_lockdown, \
scx_base_just_a_flu, sc3_loose_lockdown, sc4_rogue_citizen, sc5_rogue_neighborhood, sc6_travelers, \
sc7_nominal_lockdown_removal, sc8_innoculation, sc9_vaccination, sc10_variant
from scenario.helper.ray import launch_parallel_byday, launch_parallel_byvariant
from simulator.constants.keys import scenario_id_key, random_seed_key, draw_graph_key, ncpu_key, show_plot_key
from simulator.helper.environment import get_environment_simulation_p, get_clean_env_params
from simulator.helper.parser import get_parser
from simulator.helper.plot import chose_draw_plot
from simulator.helper.simulation import get_default_params
if __name__ == '__main__':
params = get_default_params()
args = get_parser().parse_args()
for arg in vars(args):
v = getattr(args, arg)
if arg in params and v is not None:
params[arg] = v
random.seed(params[random_seed_key])
numpy.random.seed(params[random_seed_key])
t_start = time.time()
params_env, key_env = get_clean_env_params(params)
env_file = "env_models/env_" + key_env + ".joblib"
if os.path.exists(env_file):