From 418f05f6e45841b2b8dac9a4c1ccdcccdec4af70 Mon Sep 17 00:00:00 2001 From: Soulter <905617992@qq.com> Date: Tue, 3 Mar 2026 16:06:49 +0800 Subject: [PATCH] fix: tests --- tests/agent/test_truncator.py | 49 ----------------------------------- 1 file changed, 49 deletions(-) diff --git a/tests/agent/test_truncator.py b/tests/agent/test_truncator.py index 1027643bb..c85fb7d55 100644 --- a/tests/agent/test_truncator.py +++ b/tests/agent/test_truncator.py @@ -52,18 +52,6 @@ class TestContextTruncator: assert len(result) == 3 assert result == messages - def test_fix_messages_tool_with_valid_context(self): - """Test fix_messages with tool message after user+assistant.""" - truncator = ContextTruncator() - messages = [ - self.create_message("user", "Run tool"), - self.create_message("assistant", "Running..."), - self.create_message("tool", "Tool result"), - ] - result = truncator.fix_messages(messages) - assert len(result) == 3 - assert result == messages - def test_fix_messages_tool_without_context(self): """Test fix_messages with tool message without enough context.""" truncator = ContextTruncator() @@ -74,43 +62,6 @@ class TestContextTruncator: # Tool message without context should be removed assert len(result) == 0 - def test_fix_messages_tool_with_only_one_message(self): - """Test fix_messages with tool message after only one message.""" - truncator = ContextTruncator() - messages = [ - self.create_message("user", "Hello"), - self.create_message("tool", "Tool result"), - ] - result = truncator.fix_messages(messages) - # Tool message without enough context should be removed - assert len(result) == 0 - - def test_fix_messages_multiple_tools(self): - """Test fix_messages with multiple tool messages.""" - truncator = ContextTruncator() - messages = [ - self.create_message("user", "Run tool"), - self.create_message("assistant", "Running..."), - self.create_message("tool", "Tool 1 result"), - self.create_message("tool", "Tool 2 result"), - ] - result = truncator.fix_messages(messages) - assert len(result) == 4 - assert result == messages - - def test_fix_messages_mixed_system_tool(self): - """Test fix_messages with system message and tool messages.""" - truncator = ContextTruncator() - messages = [ - self.create_message("system", "System prompt"), - self.create_message("user", "Run tool"), - self.create_message("assistant", "Running..."), - self.create_message("tool", "Tool result"), - ] - result = truncator.fix_messages(messages) - assert len(result) == 4 - assert result == messages - # ==================== truncate_by_turns Tests ==================== def test_truncate_by_turns_no_limit(self):