test(cli): cover output image export policy

Signed-off-by: Hassan Raza <raihassanraza10@gmail.com>

Author: M-Hassan-Raza

docling/cli/main.py

@@ -370,7 +370,7 @@ def _split_list(raw: str | None) -> list[str] | None:
     return re.split(r"[;,]", raw)
 
 
-_IMAGE_CAPABLE_EXPORT_FORMATS = frozenset(
+_OUTPUT_FORMATS_SUPPORTING_IMAGE_EXPORT_MODE = frozenset(
     {
         OutputFormat.JSON,
         OutputFormat.YAML,
@@ -386,7 +386,8 @@ def _should_generate_export_images(
     to_formats: list[OutputFormat],
 ) -> bool:
     return image_export_mode != ImageRefMode.PLACEHOLDER and any(
-        to_format in _IMAGE_CAPABLE_EXPORT_FORMATS for to_format in to_formats
+        to_format in _OUTPUT_FORMATS_SUPPORTING_IMAGE_EXPORT_MODE
+        for to_format in to_formats
     )
 
 

tests/test_cli.py

@@ -74,6 +74,24 @@ def test_should_generate_export_images(image_export_mode, to_formats, expected):
     assert _should_generate_export_images(image_export_mode, to_formats) is expected
 
 
+def test_image_export_policy_covers_all_output_formats():
+    image_export_formats = {
+        OutputFormat.JSON,
+        OutputFormat.YAML,
+        OutputFormat.HTML,
+        OutputFormat.HTML_SPLIT_PAGE,
+        OutputFormat.MARKDOWN,
+    }
+    non_image_export_formats = {
+        OutputFormat.TEXT,
+        OutputFormat.DOCTAGS,
+        OutputFormat.VTT,
+    }
+
+    assert image_export_formats.isdisjoint(non_image_export_formats)
+    assert image_export_formats | non_image_export_formats == set(OutputFormat)
+
+
 @pytest.mark.parametrize(
     ("to_format", "expect_generated_images"),
     [