mirror of
https://github.com/docling-project/docling.git
synced 2026-05-17 13:10:38 +00:00
Aditya Sasidhar
60fc517af0
chore:Condensing the latex test backend into multiple files Signed-off-by: Aditya Sasidhar <telikicherlaadityasasidhar@gmail.com>
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
from io import BytesIO
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
from docling_core.types.doc import DocItemLabel, GroupLabel
|
|
|
|
from docling.backend.latex_backend import LatexDocumentBackend
|
|
from docling.datamodel.backend_options import LatexBackendOptions
|
|
from docling.datamodel.base_models import InputFormat
|
|
from docling.datamodel.document import ConversionResult, DoclingDocument, InputDocument
|
|
from docling.document_converter import DocumentConverter
|
|
|
|
from ..test_data_gen_flag import GEN_TEST_DATA
|
|
from ..verify_utils import verify_document, verify_export
|
|
|
|
GENERATE = GEN_TEST_DATA
|
|
LATEX_DATA_DIR = Path("./tests/data/latex/")
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def latex_paths() -> list[Path]:
|
|
"""Find all LaTeX files in the test data directory."""
|
|
directory = Path("./tests/data/latex/")
|
|
if not directory.exists():
|
|
return []
|
|
|
|
paths = list(directory.glob("*.tex"))
|
|
|
|
for subdir in directory.iterdir():
|
|
if subdir.is_dir():
|
|
if (subdir / "main.tex").exists():
|
|
paths.append(subdir / "main.tex")
|
|
elif (subdir / f"arxiv_{subdir.name}.tex").exists():
|
|
paths.append(subdir / f"arxiv_{subdir.name}.tex")
|
|
|
|
return sorted(paths)
|