Reduce number of params in unit tests

paulovcmedeiros · Mar 3, 2024 · 709a3cb · 709a3cb
1 parent b01ed80
commit 709a3cb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -150,12 +150,12 @@ def _mock_input(*args, **kwargs):  # noqa: ARG001
     )
 
 
-@pytest.fixture(params=ChatOptions.get_allowed_values("model"))
+@pytest.fixture(params=ChatOptions.get_allowed_values("model")[:3])
 def llm_model(request):
     return request.param
 
 
-@pytest.fixture(params=ChatOptions.get_allowed_values("context_model"))
+@pytest.fixture(params=ChatOptions.get_allowed_values("context_model")[:3])
 def context_model(request):
     return request.param