From 739e9f0f011fef37f0a85b131a2ded47904edca3 Mon Sep 17 00:00:00 2001 From: Philippe Prados Date: Thu, 13 Mar 2025 16:45:43 +0100 Subject: [PATCH] Fix revue --- .../document_loaders/parsers/test_pdf_parsers.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/libs/community/tests/unit_tests/document_loaders/parsers/test_pdf_parsers.py b/libs/community/tests/unit_tests/document_loaders/parsers/test_pdf_parsers.py index 13c6a5f9981..6e5b2528cd8 100644 --- a/libs/community/tests/unit_tests/document_loaders/parsers/test_pdf_parsers.py +++ b/libs/community/tests/unit_tests/document_loaders/parsers/test_pdf_parsers.py @@ -77,23 +77,24 @@ def _assert_with_parser(parser: BaseBlobParser, *, splits_by_page: bool = True) @pytest.mark.parametrize( - "parser_class,require,params", + "parser_class,require,ctr_params,params", [ - (PDFMinerParser, "pdfminer", {"splits_by_page": False}), - (PyMuPDFParser, "pymupdf", {}), - (PyPDFParser, "pypdf", {}), - (PyPDFium2Parser, "pypdfium2", {}), + (PDFMinerParser, "pdfminer", {}, {"splits_by_page": False}), + (PyMuPDFParser, "pymupdf", {}, {}), + (PyPDFParser, "pypdf", {}, {}), + (PyPDFium2Parser, "pypdfium2", {}, {}), ], ) def test_parsers( parser_class: Type, require: str, + ctr_params: dict[str, Any], params: dict[str, Any], ) -> None: try: require = require.replace("-", "") importlib.import_module(require, package=None) - parser = parser_class() + parser = parser_class(**ctr_params) _assert_with_parser(parser, **params) except ModuleNotFoundError: pytest.skip(f"{parser_class} skiped. Require '{require}'")