add more test

This commit is contained in:
QuanyiLi
2023-05-08 16:17:27 +01:00
parent 348ee642c8
commit f610e622a8
4 changed files with 20 additions and 5 deletions

View File

@@ -0,0 +1,14 @@
"""
This script aims to convert nuscenes scenarios to ScenarioDescription, so that we can load any nuscenes scenarios into
MetaDrive.
"""
import argparse
from scenarionet.verifier.utils import verify_loading_into_metadrive
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument("--dataset_path", required=True, help="Dataset path, a directory")
parser.add_argument("--result_save_dir", required=True, help="Dataset path, a directory")
args = parser.parse_args()
verify_loading_into_metadrive(args.dataset_path, args.result_save_dir)

View File

@@ -16,8 +16,8 @@ def _test_combine_dataset():
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)
os.makedirs("verify_results", exist_ok=True)
success, result = verify_loading_into_metadrive(combine_path, "verify_results")
assert success
verify_loading_into_metadrive(combine_path, "verify_results")
# assert success
if __name__ == '__main__':

View File

@@ -0,0 +1,4 @@
#!/usr/bin/env bash
python ../../scripts/combine_dataset.py --to_dataset ../../dataset/test_combine_dataset --from_dataset ../../dataset/waymo ../../dataset/pg ../../dataset/nuscenes ../../dataset/nuplan --overwrite
python ../../scripts/combine_dataset.py --dataset_path ../../dataset/test_combine_dataset --result_save_dir ../../dataset/test_combine_dataset

View File

@@ -1,3 +0,0 @@
#!/usr/bin/env bash
nohup python ../../scripts/combine_dataset.py --to_dataset ../../dataset/test_combine_dataset --from_dataset ../../dataset/waymo ../../dataset/pg ../../dataset/nuscenes ../../dataset/nuplan --overwrite > nuplan.log 2>&1 &