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