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'