def test_dict_paramss(dict_of_params): sg = ScenarioGenerator() sg.parameters = dict_of_params print(sg.parameters) sg._handle_input_parameters() assert len(sg.all_permutations) == 6
def test_folder_creation(tmpdir): sg = ScenarioGenerator() # test no folder existing nodir = os.path.join(tmpdir, 'dir1') sg._create_folder_structure(nodir) assert 'xodr' in os.listdir(nodir) assert 'xosc' in os.listdir(nodir) # test path exist dir_exist = os.path.join(tmpdir, 'dir2') os.mkdir(dir_exist) sg._create_folder_structure(dir_exist) assert 'xodr' in os.listdir(dir_exist) assert 'xosc' in os.listdir(dir_exist) # test folder exists and the xosc fodler xosc_dir_exist = os.path.join(tmpdir, 'dir3') os.mkdir(xosc_dir_exist) os.mkdir(os.path.join(xosc_dir_exist, 'xosc')) sg._create_folder_structure(xosc_dir_exist) assert 'xodr' in os.listdir(xosc_dir_exist) assert 'xosc' in os.listdir(xosc_dir_exist) # test folder exists and the xodr fodler xodr_dir_exist = os.path.join(tmpdir, 'dir4') os.mkdir(xodr_dir_exist) os.mkdir(os.path.join(xodr_dir_exist, 'xodr')) sg._create_folder_structure(xodr_dir_exist) assert 'xodr' in os.listdir(xodr_dir_exist) assert 'xosc' in os.listdir(xodr_dir_exist)
def __init__(self): ScenarioGenerator.__init__(self) self.parameters = [] d1 = {} d1['road_curvature'] = 0.001 d1['speed'] = 10 d2 = {} d2['road_curvature'] = 0.002 d2['speed'] = 20 self.parameters.append(d1) self.parameters.append(d2) self.naming = 'parameter'
def test_parameter_naming(list_of_params): sg = ScenarioGenerator() sg.parameters = list_of_params sg.naming = 'parameter' name = sg._get_scenario_name(sg.parameters[0]) assert name == "scenario_generator_road_curvature-0.001_speed-10" name = sg._get_scenario_name(sg.parameters[1]) assert name == "scenario_generator_road_curvature-0.002_speed-20"
def test_numerical_naming(list_of_params): sg = ScenarioGenerator() sg.parameters = list_of_params sg.naming = 'numerical' name = sg._get_scenario_name(sg.parameters[0]) assert name == "scenario_generator0" name = sg._get_scenario_name(sg.parameters[1]) assert name == "scenario_generator1"
def __init__(self): ScenarioGenerator.__init__(self) self.naming = 'numerical'
def test_list_of_params(list_of_params): sg = ScenarioGenerator() sg.parameters = list_of_params sg._handle_input_parameters() assert len(sg.all_permutations) == 2
def __init__(self, parameters, naming): ScenarioGenerator.__init__(self) self.parameters = parameters self.naming = naming
def __init__(self): ScenarioGenerator.__init__(self) self.parameters['road_curvature'] = [0.001, 0.002, 0.003, 0.004] self.parameters['speed'] = [10, 20, 30] self.naming = 'numerical'