File size: 505 Bytes
170d9a9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import pytest


def pytest_addoption(parser):
    parser.addoption(
        "--model-name",
        action="store",
        default="gemini-1.5-flash",
        help="Model name to use for evaluation",
    )


def pytest_configure(config):
    # Add model_name to pytest namespace for access in tests
    config.addinivalue_line(
        "markers", "model_name: mark test to run with specific model name"
    )


@pytest.fixture
def model_name(request):
    return request.config.getoption("--model-name")