diff --git a/tests/test_four_phase_unit.py b/tests/test_four_phase_unit.py index b7f723d..d7df249 100644 --- a/tests/test_four_phase_unit.py +++ b/tests/test_four_phase_unit.py @@ -213,7 +213,6 @@ class TestU3RoundComplete: ticker._build_review_prompt = MagicMock(return_value="Review prompt mock") spawner=mock_spawner, dispatcher=MagicMock()) # mock _build_review_prompt(f-string 在 Python 3.9 有兼容性问题) - ticker._build_review_prompt = MagicMock(return_value="Review prompt mock") async def run(): result = await ticker._check_round_complete(db_path, pid) @@ -243,6 +242,7 @@ class TestU3RoundComplete: ticker = Ticker(registry=MagicMock(), spawner=MagicMock(), dispatcher=MagicMock()) + ticker._build_review_prompt = MagicMock(return_value="Review prompt mock") async def run(): return await ticker._check_round_complete(db_path, pid) @@ -313,6 +313,7 @@ class TestU5RoundLimit: ticker = Ticker(registry=MagicMock(), spawner=MagicMock(), dispatcher=MagicMock()) + ticker._build_review_prompt = MagicMock(return_value="Review prompt mock") async def run(): return await ticker._check_round_complete(db_path, pid) @@ -383,6 +384,7 @@ class TestU6MentionGrouping: ticker = Ticker(registry=MagicMock(), spawner=mock_spawner, dispatcher=MagicMock()) + ticker._build_review_prompt = MagicMock(return_value="Review prompt mock") async def run(): return await ticker._process_mentions(db_path, pid)