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

    if 'basic_linargs' == metafunc.fixturenames:
        fargs = []
        if metafunc.config.option.all:
            nin_rng = [1, 2, 64, 128]
            nout_rng = [1, 4, 128, 64]
            vocab_size = [1, 4, 1000, 2000]
        else:
            nin_rng = [4, 32]
            nout_rng = [3, 33]
            vocab_size = [10, 34]
        fargs = itt.product(nin_rng, nout_rng, vocab_size, bsz_rng)
        print fargs
        metafunc.parametrize('basic_linargs', fargs)
dataset/ETHPy150Open NervanaSystems/neon/tests/test_lookuptable.py/pytest_generate_tests/operator `in` -> `==`