libs/kotaemon/tests/_test_multimodal_reader.py

# TODO: This test is broken and should be rewritten
from pathlib import Path

from kotaemon.loaders import AdobeReader

# from dotenv import load_dotenv


input_file = Path(__file__).parent / "resources" / "multimodal.pdf"

# load_dotenv()


def test_adobe_reader():
    reader = AdobeReader()
    documents = reader.load_data(input_file)
    table_docs = [doc for doc in documents if doc.metadata.get("type", "") == "table"]
    assert len(table_docs) == 2

    figure_docs = [doc for doc in documents if doc.metadata.get("type", "") == "image"]
    assert len(figure_docs) == 2