from operator import itemgetter

import pytest


@pytest.fixture(params=[220, 30])
def page_height(request):
    return request.param


@pytest.fixture(params=[100, 310])
def page_width(request):
    return request.param


@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


@pytest.fixture(params=[0, 1, 2, 16, 32])
def batch_size(request):
    return request.param