Avoid test case remove the example file

Microwave-WYB · May 21, 2024 · e73f8e9 · e73f8e9
1 parent 3ae91c4
commit e73f8e9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/phone_sensors/birdnet.py b/phone_sensors/birdnet.py
@@ -18,7 +18,7 @@
 
 
 def analyze_audio(
-    file_path: FilePath, sensor_status: SensorStatus, min_conf: float
+    file_path: FilePath, sensor_status: SensorStatus, min_conf: float, remove_file: bool = True
 ) -> tuple[list[BirdNetDetection], SensorStatus]:
     """Analyze audio file and return list of bird species."""
     logger.info("Analyzing audio file %s...", file_path)
@@ -32,7 +32,8 @@ def analyze_audio(
     )
     recording.analyze()
     detections = [BirdNetDetection.model_validate(d) for d in recording.detections]
-    file_path.unlink()
+    if remove_file:
+        file_path.unlink()
     return detections, sensor_status
 
 

diff --git a/tests/test_birdnet.py b/tests/test_birdnet.py
@@ -72,6 +72,6 @@ def test_analyze_audio():
         battery=0.5,
         temperature=20.0,
     ).add_coordinates()
-    result = analyze_audio(file_path, sensor_status, 0.25)
+    result = analyze_audio(file_path, sensor_status, 0.25, remove_file=False)
     rich.print(result)
     assert len(result) > 0