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)