def pytest_generate_tests(metafunc):
    np.random.seed(1)
    if metafunc.config.option.all:
        bsz_rng = [32, 64]
    else:
        bsz_rng = [128]

    if 'poolargs' in metafunc.fixturenames:
        # to check padding, do not need large input
        fargs = []
        if metafunc.config.option.all:
            fs_rng = [2, 3, 5]
            pad_rng = [0, 1]
            nifm_rng = [16, 32]
            in_sz_rng = [8, 16]
        else:
            fs_rng = [2, 4]
            pad_rng = [0, 1]
            nifm_rng = [8]
            in_sz_rng = [8]
        fargs_ = []
        for fs in fs_rng:
            stride_rng = set([1, fs/2, fs])
            fargs_.append(itt.product(fs_rng, nifm_rng, pad_rng, stride_rng, in_sz_rng, bsz_rng))
        fargs = itt.chain(*fargs_)
        metafunc.parametrize('poolargs', fargs)
dataset/ETHPy150Open NervanaSystems/neon/tests/test_pool_layer.py/pytest_generate_tests/original