28 lines
920 B
Python
28 lines
920 B
Python
import os
|
||
|
||
# 设置虚拟 key 以避免测试收集期间 pydantic-ai 初始化错误
|
||
os.environ["DEEPSEEK_API_KEY"] = "dummy_key_for_testing"
|
||
|
||
import sys
|
||
sys.path.append(os.getcwd())
|
||
|
||
from src.agent_app import predict_student
|
||
|
||
# 注意: 我们直接测试工具函数，而不是完整的 agent 循环
|
||
# 因为 agent 需要 API key，而 CI/测试环境中可能未设置。
|
||
|
||
from unittest.mock import patch
|
||
|
||
def test_tool_wrapper():
|
||
# 测试 Agent wrapper 函数是否能正确调用到底层 infer
|
||
# 我们 mock 底层的 predict_pass_prob，这样测试就不依赖于实际的模型文件是否存在
|
||
with patch("src.agent_app.predict_pass_prob") as mock_predict:
|
||
mock_predict.return_value = 0.85
|
||
|
||
prob = predict_student(None, 12, 8, 0.9, 2, "Self")
|
||
|
||
# 验证调用
|
||
assert prob == 0.85
|
||
mock_predict.assert_called_once_with(12, 8, 0.9, 2, "Self")
|
||
|