tests/conftest.py

from pathlib import Path

import PIL.Image
import pytest
from dotenv import load_dotenv
from PIL.Image import Image

load_dotenv()
asset_dir = Path(__file__).parent / 'assets'


@pytest.fixture
def image() -> Image:
    return PIL.Image.open(asset_dir / 'sample.jpeg')


@pytest.fixture
def pdf() -> bytes:
    with open(asset_dir / 'sample.pdf', 'rb') as file:
        return file.read()


@pytest.fixture
def image_path() -> str:
    return str(asset_dir / 'sample.jpeg')


@pytest.fixture
def pdf_path() -> str:
    return str(asset_dir / 'sample.pdf')