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)
def test_build_environment_dic(self):
params_test = {
nindividual_key: 10,
store_per_house_key: 2,
store_preference_key: 1,
nb_1d_block_key: 5,
remote_work_key: 0.5,
store_nb_choice_key: 3,
transport_contact_cap_key: 10
}
result = get_environment_simulation(params_test)
self.assertTrue(IH_K in result.keys())
self.assertTrue(HI_K in result.keys())
self.assertTrue(IAD_K in result.keys())
self.assertTrue(IAG_K in result.keys())
self.assertTrue(IW_K in result.keys())
self.assertTrue(WI_K in result.keys())
self.assertTrue(HA_K in result.keys())
self.assertTrue(HS_K in result.keys())
self.assertTrue(ITI_K in result.keys())
self.assertTrue(IDEA_K in result.keys())
self.assertTrue(IHOS_K in result.keys())
import random
import time
from scenario.example import sc0_base_lockdown
from simulator.constants.keys import random_seed_key, draw_graph_key
from simulator.helper.parser import get_parser
from simulator.helper.plot import chose_draw_plot
from simulator.helper.simulation import get_default_params
from simulator.helper.environment import get_environment_simulation
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])
env_dic = get_environment_simulation(params)
t_start = time.time()
stats_result = sc0_base_lockdown.launch_run(params, env_dic)
print("It took : %.2f seconds" % (time.time() - t_start))
chose_draw_plot(params[draw_graph_key], stats_result)
def test_build_environment_dic(self):
params_test = {
nindividual_key: 10,
store_per_house_key: 2,
store_preference_key: 1,
nb_1d_block_key: 5,
remote_work_key: 0.5,
store_nb_choice_key: 3,
transport_contact_cap_key: 10
}
result = get_environment_simulation(params_test)
expected_result = {
IH_K: {
0: 0,
1: 0,
2: 1,
3: 1,
4: 1,
5: 1,
6: 1,
7: 2,
8: 2,
9: 2
},
HI_K: {
0: [0, 1],
1: [2, 3, 4, 5, 6],
2: [7, 8, 9]
},
IAD_K: {
0: 1,
1: 1,
2: 1,
3: 1,
4: 0,
5: 0,
6: 0,
7: 1,
8: 1,
9: 0
},
IAG_K: {
0: 35,
1: 47,
2: 33,
3: 23,
4: 2,
5: 15,
6: 13,
7: 26,
8: 37,
9: 0
},
IW_K: {
0: 0,
1: 0,
2: 0,
3: 0,
7: 0,
8: 0
},
WI_K: {
0: [0, 3, 7, 2, 1, 8]
},
HA_K: {
0: [0, 1],
1: [2, 3],
2: [7, 8]
},
HS_K: {
0: [0, 1, 1],
1: [0, 1, 1],
2: [0, 1, 1]
},
ITI_K: {
0: {0, 1, 2, 3, 7, 8},
1: {0, 1, 2, 3, 7, 8},
2: {0, 1, 2, 3, 7, 8},
3: {0, 1, 2, 3, 7, 8},
7: {0, 1, 2, 3, 7, 8},
8: {0, 1, 2, 3, 7, 8}
}
}
self.assertTrue(IBE_K in result.keys())
self.assertTrue(HB_K in result.keys())
# We delete them in order to avoid including 100 values dictionnary in the test
# we verified they exists it should be enough
del result[IBE_K]
del result[HB_K]
self.assertEqual(result, expected_result)