36 lines
748 B
Python
36 lines
748 B
Python
from operator import itemgetter
|
|
|
|
import pytest
|
|
|
|
from test.utils.generation.image import array_to_image
|
|
|
|
|
|
@pytest.fixture
|
|
def images(input_batch):
|
|
return list(map(array_to_image, input_batch))
|
|
|
|
|
|
@pytest.fixture
|
|
def input_size(alpha, __input_size):
|
|
w, h, d = __input_size
|
|
return w, h, d + alpha
|
|
|
|
|
|
@pytest.fixture(params=[False])
|
|
def alpha(request):
|
|
return request.param
|
|
|
|
|
|
@pytest.fixture(params=[{"width": 10, "height": 15, "depth": 3}, {"width": 150, "height": 100, "depth": 3}])
|
|
def __input_size(request):
|
|
return itemgetter("width", "height", "depth")(request.param)
|
|
|
|
|
|
@pytest.fixture(params=[33, 100])
|
|
def height(request):
|
|
return request.param
|
|
|
|
|
|
@pytest.fixture(params=[10, 31])
|
|
def width(request):
|
|
return request.param |