multi-processing test

2023-05-07 23:14:01 +01:00
parent 5c1b2e053b
commit 97a4e61f38
4 changed files with 24 additions and 21 deletions
--- a/scenarionet/tests/test_combine_dataset.py
+++ b/scenarionet/tests/test_combine_dataset.py
@@ -21,9 +21,11 @@ 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))
+        num_worker = 4 if len(summary) > 4 else 1
        success, result = verify_loading_into_metadrive(dataset_path,
                                                        result_save_dir="test_dataset",
-                                                        steps_to_run=300)
+                                                        steps_to_run=1000,
+                                                        num_workers=num_worker)
        assert success