Mirrors/kotaemon
1
0
Fork 0
mirror of https://github.com/Cinnamon/kotaemon.git synced 2026-08-29 10:08:59 +02:00
Code Issues Packages Projects Releases Wiki Activity

tests: add skip_when_paddleocr_not_installed

Browse Source
This commit is contained in:
cin-niko
2026-03-08 06:00:17 +00:00
parent b8c18ad98c
commit 1470174f20
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
libs/kotaemon/tests/test_paddleocr_loader.py
View File

@@ -110,6 +110,7 @@ def test_to_documents_skips_empty_content() -> None:
# --------------------------------------------------------------------------- # ---------------------------------------------------------------------------
@skip_when_paddleocr_not_installed
def test_ppstructure_v3_unsupported_file_type_raises() -> None: def test_ppstructure_v3_unsupported_file_type_raises() -> None:
"""load_data raises ValueError for unsupported extension.""" """load_data raises ValueError for unsupported extension."""
from kotaemon.loaders import PPStructureV3Reader from kotaemon.loaders import PPStructureV3Reader
@@ -129,6 +130,7 @@ def test_ppstructure_v3_supported_file_types_attribute() -> None:
assert ".png" in reader.supported_file_types assert ".png" in reader.supported_file_types
@skip_when_paddleocr_not_installed
def test_ppstructure_v3_load_data_with_mocked_pipeline( def test_ppstructure_v3_load_data_with_mocked_pipeline(
mocker: MockerFixture, mocker: MockerFixture,
) -> None: ) -> None:
@@ -170,6 +172,7 @@ def test_ppstructure_v3_load_data_with_mocked_pipeline(
# --------------------------------------------------------------------------- # ---------------------------------------------------------------------------
@skip_when_paddleocr_not_installed
def test_paddleocr_vl_unsupported_file_type_raises() -> None: def test_paddleocr_vl_unsupported_file_type_raises() -> None:
"""load_data raises ValueError for unsupported extension.""" """load_data raises ValueError for unsupported extension."""
from kotaemon.loaders import PaddleOCRVLReader from kotaemon.loaders import PaddleOCRVLReader
@@ -189,6 +192,7 @@ def test_paddleocr_vl_supported_file_types_attribute() -> None:
assert ".png" in reader.supported_file_types assert ".png" in reader.supported_file_types
@skip_when_paddleocr_not_installed
def test_paddleocr_vl_load_data_with_mocked_pipeline( def test_paddleocr_vl_load_data_with_mocked_pipeline(
mocker: MockerFixture, mocker: MockerFixture,
) -> None: ) -> None: