trpc-agent-python/tests/sessions/conftest.py at a9df99269bbf8f04d2d8fc1e5e6c71f8dd17b0b9 · trpc-group/trpc-agent-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
# Tencent is pleased to support the open source community by making tRPC-Agent-Python available.
#
# Copyright (C) 2026 Tencent. All rights reserved.
#
# tRPC-Agent-Python is licensed under Apache-2.0.
"""Pytest fixtures for replay consistency testing.

每个 fixture 对应一个 replay case，测试类通过 fixture 名称引用。
"""

from __future__ import annotations

import pytest

from .replay_cases import (
    ReplayCase,
    build_single_turn_text,
    build_multi_turn_text,
    build_tool_call_response,
    build_state_basic_update,
    build_state_three_tier,
    build_memory_store_search,
    build_memory_multi_session,
    build_summary_generation,
    build_summary_truncation,
    build_summary_error_detection,
    build_error_duplicate_write,
)


@pytest.fixture(scope="module")
def case_single_turn_text() -> ReplayCase:
    return build_single_turn_text()


@pytest.fixture(scope="module")
def case_multi_turn_text() -> ReplayCase:
    return build_multi_turn_text()


@pytest.fixture(scope="module")
def case_tool_call_response() -> ReplayCase:
    return build_tool_call_response()


@pytest.fixture(scope="module")
def case_state_basic_update() -> ReplayCase:
    return build_state_basic_update()


@pytest.fixture(scope="module")
def case_state_three_tier() -> ReplayCase:
    return build_state_three_tier()


@pytest.fixture(scope="module")
def case_memory_store_search() -> ReplayCase:
    return build_memory_store_search()


@pytest.fixture(scope="module")
def case_memory_multi_session() -> ReplayCase:
    return build_memory_multi_session()


@pytest.fixture(scope="module")
def case_summary_generation() -> ReplayCase:
    return build_summary_generation()


@pytest.fixture(scope="module")
def case_summary_truncation() -> ReplayCase:
    return build_summary_truncation()


@pytest.fixture(scope="module")
def case_summary_error_detection() -> ReplayCase:
    return build_summary_error_detection()


@pytest.fixture(scope="module")
def case_error_duplicate_write() -> ReplayCase:
    return build_error_duplicate_write()