28 lines
595 B
Python
28 lines
595 B
Python
import os
|
|
|
|
import fpdf
|
|
import pytest
|
|
|
|
from image_prediction.locations import TEST_DATA_DIR
|
|
from test.utils.generation.pdf import add_image, pdf_stream, stream_pdf_bytes
|
|
|
|
|
|
@pytest.fixture
|
|
def pdf(image_metadata_pairs):
|
|
pdf = fpdf.FPDF(unit="pt")
|
|
|
|
for pair in image_metadata_pairs:
|
|
add_image(pdf, pair)
|
|
|
|
return pdf_stream(pdf)
|
|
|
|
|
|
@pytest.fixture
|
|
def real_pdf(dvc_test_data):
|
|
yield from stream_pdf_bytes(TEST_DATA_DIR / "f2dc689ca794fccb8cd38b95f2bf6ba9.pdf")
|
|
|
|
|
|
@pytest.fixture
|
|
def bad_xref_pdf(dvc_test_data):
|
|
yield from stream_pdf_bytes(TEST_DATA_DIR / "bad_xref.pdf")
|