Adjust chunksize for tests

cta-observatory · Nov 23, 2022 · 2c2bcc7 · 2c2bcc7
1 parent 323c603
commit 2c2bcc7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ctapipe/tools/tests/test_apply_models.py b/ctapipe/tools/tests/test_apply_models.py
@@ -26,6 +26,7 @@ def test_apply_energy_regressor(
             f"--output={output_path}",
             f"--energy-regressor={energy_regressor_path}",
             "--StereoMeanCombiner.weights=konrad",
+            "--chunk-size=5",  # small chunksize so we test multiple chunks for the test file
         ],
         raises=True,
     )
@@ -86,6 +87,7 @@ def test_apply_both(
             f"--energy-regressor={energy_regressor_path}",
             f"--particle-classifier={particle_classifier_path}",
             "--StereoMeanCombiner.weights=konrad",
+            "--chunk-size=5",  # small chunksize so we test multiple chunks for the test file
         ],
     )
     assert ret == 0