Handle case where test host has jpeginfo not installed

zauberzeug · Jan 16, 2024 · 0d045c8 · 0d045c8
1 parent 8acb75b
commit 0d045c8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/learning_loop_node/tests/test_downloader.py b/learning_loop_node/tests/test_downloader.py
@@ -2,7 +2,7 @@
 import shutil
 
 from learning_loop_node.data_classes import Context
-from learning_loop_node.data_exchanger import DataExchanger
+from learning_loop_node.data_exchanger import DataExchanger, check_jpeg
 from learning_loop_node.globals import GLOBALS
 
 from . import test_helper
@@ -67,5 +67,5 @@ async def test_removal_of_corrupted_images(data_exchanger: DataExchanger):
 
     await data_exchanger.delete_corrupt_images('/tmp/img_folder')
 
-    assert len(os.listdir('/tmp/img_folder')) == num_images
+    assert len(os.listdir('/tmp/img_folder')) == num_images if check_jpeg else num_images - 1
     shutil.rmtree('/tmp/img_folder', ignore_errors=True)