mirror of
https://github.com/jihe520/MathModelAgent.git
synced 2026-09-19 08:09:48 +08:00
jihe520
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
import pytest
|
|
from utils.RichPrinter import RichPrinter
|
|
|
|
|
|
def test_rich_printer_basic():
|
|
"""测试 RichPrinter 的基本功能"""
|
|
|
|
# 测试工作流开始和结束
|
|
RichPrinter.workflow_start()
|
|
RichPrinter.workflow_end()
|
|
|
|
# 测试不同类型的消息打印
|
|
RichPrinter.success("这是一条成功消息")
|
|
RichPrinter.error("这是一条错误消息")
|
|
RichPrinter.warning("这是一条警告消息")
|
|
|
|
# 测试带自定义参数的消息
|
|
RichPrinter.success("自定义标题的成功消息", title="自定义标题")
|
|
RichPrinter.error("自定义颜色的错误消息", color="bright_red")
|
|
|
|
# 测试表格打印
|
|
headers = ["编号", "姓名", "得分"]
|
|
rows = [[1, "张三", 95], [2, "李四", 88], [3, "王五", 92]]
|
|
RichPrinter.table(headers=headers, rows=rows, title="学生成绩表")
|
|
|
|
# 测试 Agent 消息
|
|
RichPrinter.print_agent_msg("正在生成代码...", "CoderAgent")
|
|
RichPrinter.print_agent_msg("正在撰写论文...", "WriterAgent")
|
|
|
|
|
|
def test_rich_printer_error_cases():
|
|
"""测试错误情况"""
|
|
|
|
# 测试未知 Agent 名称
|
|
with pytest.raises(ValueError):
|
|
RichPrinter.print_agent_msg("测试消息", "UnknownAgent")
|