test: Parameterize DeepSeek model component test correctly (#7019)

test: Parameterize DeepSeek model component test with temperature and max tokens
This commit is contained in:
Gabriel Luiz Freitas Almeida 2025-03-12 21:25:32 -03:00 • committed by GitHub
commit 979c4a6d70
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -40,6 +40,11 @@ def test_deepseek_template():
assert input_name in input_names assert input_name in input_names
@pytest.fixture
def mock_chat_openai(mocker):
return mocker.patch("langchain_openai.ChatOpenAI")
@pytest.mark.parametrize( @pytest.mark.parametrize(
("temperature", "max_tokens"), ("temperature", "max_tokens"),
[ [
@ -48,15 +53,10 @@ def test_deepseek_template():
(1.5, 1000), (1.5, 1000),
], ],
) )
@pytest.fixture def test_deepseek_build_model(mock_chat_openai, temperature, max_tokens):
def mock_chat_openai(mocker):
return mocker.patch("langchain_openai.ChatOpenAI")
def test_deepseek_build_model(mock_chat_openai):
component = DeepSeekModelComponent() component = DeepSeekModelComponent()
component.temperature = 0.7 component.temperature = temperature
component.max_tokens = 100 component.max_tokens = max_tokens
component.api_key = "test-key" component.api_key = "test-key"
# Mock the ChatOpenAI instance # Mock the ChatOpenAI instance
@ -67,12 +67,12 @@ def test_deepseek_build_model(mock_chat_openai):
# Verify ChatOpenAI was called with correct params # Verify ChatOpenAI was called with correct params
mock_chat_openai.assert_called_once_with( mock_chat_openai.assert_called_once_with(
max_tokens=100, max_tokens=max_tokens,
model_kwargs={}, model_kwargs={},
model="deepseek-chat", model="deepseek-chat",
base_url="https://api.deepseek.com", base_url="https://api.deepseek.com",
api_key="test-key", api_key="test-key",
temperature=0.7, temperature=temperature,
seed=1, seed=1,
streaming=False, streaming=False,
) )