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

add filter test

Browse Source
This commit is contained in:
QuanyiLi
2023-05-07 20:41:46 +01:00
parent 805841ac93
commit 468cff9040
1 changed files with 14 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
scenarionet/tests/test_filter.py
View File

@@ -15,7 +15,7 @@ def test_filter_dataset():
output_path = os.path.join(SCENARIONET_PACKAGE_PATH, "tests", "combine") output_path = os.path.join(SCENARIONET_PACKAGE_PATH, "tests", "combine")
# test 1 # ========================= test 1 =========================
# nuscenes data has no light # nuscenes data has no light
# light_condition = ScenarioFilter.make(ScenarioFilter.has_traffic_light) # light_condition = ScenarioFilter.make(ScenarioFilter.has_traffic_light)
sdc_driving_condition = ScenarioFilter.make(ScenarioFilter.sdc_moving_dist, sdc_driving_condition = ScenarioFilter.make(ScenarioFilter.sdc_moving_dist,
@@ -36,12 +36,23 @@ def test_filter_dataset():
break break
assert in_ assert in_
# test 2 # ========================= test 2 =========================
num_condition = ScenarioFilter.make(ScenarioFilter.object_number, num_condition = ScenarioFilter.make(ScenarioFilter.object_number,
number_threshold=6, number_threshold=50,
object_type=MetaDriveType.PEDESTRIAN, object_type=MetaDriveType.PEDESTRIAN,
condition="greater") condition="greater")
answer = ['sd_nuscenes_v1.0-mini_scene-0061.pkl', 'sd_nuscenes_v1.0-mini_scene-1094.pkl']
summary, mapping = combine_multiple_dataset(output_path,
*dataset_paths,
force_overwrite=True,
try_generate_missing_file=True,
filters=[num_condition])
assert len(answer) == len(summary)
for a in answer:
assert a in summary
if __name__ == '__main__': if __name__ == '__main__':
test_filter_dataset() test_filter_dataset()