huangfu/scenarionet
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test scenarios

Browse Source
This commit is contained in:
QuanyiLi
2023-05-07 16:05:42 +01:00
parent 725cc7d2b9
commit b0791e5136
5 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scenarionet/examples/run_combine_scenarios.py
View File

@@ -11,7 +11,7 @@ if __name__ == '__main__':
dataset_paths = [os.path.join(SCENARIONET_DATASET_PATH, "nuscenes")]
dataset_paths.append(os.path.join(SCENARIONET_DATASET_PATH, "nuplan"))
dataset_paths.append(os.path.join(SCENARIONET_DATASET_PATH, "waymo"))
dataset_paths.append(os.path.join(SCENARIONET_DATASET_PATH, "nuscenes"))
dataset_paths.append(os.path.join(SCENARIONET_DATASET_PATH, "pg"))
combine_path = os.path.join(SCENARIONET_DATASET_PATH, "combined_dataset")
combine_multiple_dataset(combine_path, *dataset_paths, force_overwrite=True, try_generate_missing_file=True)

2
scenarionet/tests/_test_combine_dataset_local.py
View File

@@ -9,7 +9,7 @@ def _test_combine_dataset():
dataset_paths = [os.path.join(SCENARIONET_DATASET_PATH, "nuscenes")]
dataset_paths.append(os.path.join(SCENARIONET_DATASET_PATH, "nuplan"))
dataset_paths.append(os.path.join(SCENARIONET_DATASET_PATH, "waymo"))
dataset_paths.append(os.path.join(SCENARIONET_DATASET_PATH, "nuscenes"))
dataset_paths.append(os.path.join(SCENARIONET_DATASET_PATH, "pg"))
combine_path = os.path.join(SCENARIONET_DATASET_PATH, "combined_dataset")
combine_multiple_dataset(combine_path, *dataset_paths, force_overwrite=True, try_generate_missing_file=True)

2
scenarionet/tests/script/run_env.py
View File

@@ -11,7 +11,7 @@ if __name__ == '__main__':
dataset_paths = [os.path.join(SCENARIONET_DATASET_PATH, "nuscenes")]
dataset_paths.append(os.path.join(SCENARIONET_DATASET_PATH, "nuplan"))
dataset_paths.append(os.path.join(SCENARIONET_DATASET_PATH, "waymo"))
dataset_paths.append(os.path.join(SCENARIONET_DATASET_PATH, "nuscenes"))
dataset_paths.append(os.path.join(SCENARIONET_DATASET_PATH, "pg"))
combine_path = os.path.join(SCENARIONET_DATASET_PATH, "combined_dataset")
combine_multiple_dataset(combine_path, *dataset_paths, force_overwrite=True, try_generate_missing_file=True)

4
scenarionet/tests/test_combine_dataset.py
View File

@@ -21,7 +21,9 @@ def test_combine_multiple_dataset():
summary, sorted_scenarios, mapping = read_dataset_summary(dataset_path)
for scenario_file in sorted_scenarios:
read_scenario(os.path.join(dataset_path, mapping[scenario_file], scenario_file))
success, result = verify_loading_into_metadrive(dataset_path, result_save_dir="./test_dataset")
success, result = verify_loading_into_metadrive(dataset_path,
result_save_dir="./test_dataset",
steps_to_run=300)
assert success

1
scenarionet/verifier/utils.py
View File

@@ -10,6 +10,7 @@ from metadrive.scenario.utils import get_number_of_scenarios
def verify_loading_into_metadrive(dataset_path, result_save_dir=None, steps_to_run=0):
print("================ Begin Scenario Loading oVerification ================ \n")
scenario_num = get_number_of_scenarios(dataset_path)
if result_save_dir is not None:
assert os.path.exists(result_save_dir) and os.path.isdir(