github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-25 04:23:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix revue

Browse Source
This commit is contained in:
Philippe Prados 2025-03-13 16:45:43 +01:00
parent c032aa7499
commit 739e9f0f01
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
libs/community/tests/unit_tests/document_loaders/parsers/test_pdf_parsers.py
View File

@ -77,23 +77,24 @@ def _assert_with_parser(parser: BaseBlobParser, *, splits_by_page: bool = True)
@pytest.mark.parametrize( @pytest.mark.parametrize(
"parser_class,require,params", "parser_class,require,ctr_params,params",
[ [
(PDFMinerParser, "pdfminer", {"splits_by_page": False}), (PDFMinerParser, "pdfminer", {}, {"splits_by_page": False}),
(PyMuPDFParser, "pymupdf", {}), (PyMuPDFParser, "pymupdf", {}, {}),
(PyPDFParser, "pypdf", {}), (PyPDFParser, "pypdf", {}, {}),
(PyPDFium2Parser, "pypdfium2", {}), (PyPDFium2Parser, "pypdfium2", {}, {}),
], ],
) )
def test_parsers( def test_parsers(
parser_class: Type, parser_class: Type,
require: str, require: str,
ctr_params: dict[str, Any],
params: dict[str, Any], params: dict[str, Any],
) -> None: ) -> None:
try: try:
require = require.replace("-", "") require = require.replace("-", "")
importlib.import_module(require, package=None) importlib.import_module(require, package=None)
parser = parser_class() parser = parser_class(**ctr_params)
_assert_with_parser(parser, **params) _assert_with_parser(parser, **params)
except ModuleNotFoundError: except ModuleNotFoundError:
pytest.skip(f"{parser_class} skiped. Require '{require}'") pytest.skip(f"{parser_class} skiped. Require '{require}'")