add test

pyronear · Jun 24, 2024 · c862307 · c862307
1 parent 1c9fa41
commit c862307
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_engine.py b/tests/test_engine.py
@@ -44,7 +44,7 @@ def test_engine_offline(tmpdir_factory, mock_wildfire_image, mock_forest_image):
     engine.clear_cache()
 
     # inference
-    engine = Engine(nb_consecutive_frames=4, cache_folder=folder)
+    engine = Engine(nb_consecutive_frames=4, cache_folder=folder, save_captured_frames=True)
     out = engine.predict(mock_forest_image)
     assert isinstance(out, float) and 0 <= out <= 1
     assert len(engine._states["-1"]["last_predictions"]) == 1