data-query-agent/tests/test_commands.py

from data_query_agent.commands import parse_command


def test_commands_are_exact_matches() -> None:
    assert parse_command(" /new ") == "new"
    assert parse_command("开启新会话") == "new"
    assert parse_command("CLEAR") == "clear"
    assert parse_command("帮助") == "help"
    assert parse_command("/skills") == "skills"
    assert parse_command("目前你有哪些skill 都支持查询什么") == "skills"
    assert parse_command("你能查询什么") == "skills"
    assert parse_command("new 查询昨天 DAU") is None