import pytest


def pytest_addoption(parser):
    parser.addoption(
        "--trials", action="store", default=10000,
        help="number of trails for random tests"
    )
    parser.addoption(
        "--dataset_samples", action="store", default=10000,
        help="number of samples to test from the data sets"
    )


@pytest.fixture
def trials(request):
    return int(request.config.getoption("--trials"))


@pytest.fixture
def dataset_samples(request):
    return int(request.config.getoption("--dataset_samples"))