25 lines
557 B
Python
25 lines
557 B
Python
import os
|
|
from typing import Callable
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture
|
|
def current_dir(request) -> str:
|
|
return os.path.dirname(request.module.__file__)
|
|
|
|
|
|
@pytest.fixture
|
|
def get_test_image(current_dir) -> Callable[[str], bytes]:
|
|
def f(name):
|
|
return open(os.path.join(current_dir, "test_images", name), "rb").read()
|
|
|
|
return f
|
|
|
|
|
|
@pytest.fixture
|
|
def save_test_image(current_dir) -> Callable[[str, bytes], None]:
|
|
def f(name: str, data: bytes):
|
|
open(os.path.join(current_dir, "test_images", name), "wb").write(data)
|
|
|
|
return f
|