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