From 63c2a98f3c499cb2820966411ef840ca91ea25dc Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Wed, 12 Mar 2025 13:41:10 -0700 Subject: [PATCH] fix: Add missing summarizer mock to ArchitectCoder tests --- tests/basic/test_coder.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/basic/test_coder.py b/tests/basic/test_coder.py index 0c13700d1..793d92a62 100644 --- a/tests/basic/test_coder.py +++ b/tests/basic/test_coder.py @@ -1075,6 +1075,8 @@ This command will print 'Hello, World!' to the console.""" coder.total_cost = 0 coder.cur_messages = [] coder.done_messages = [] + coder.summarizer = MagicMock() + coder.summarizer.too_big.return_value = False # Mock editor_coder creation and execution mock_editor = MagicMock() @@ -1108,8 +1110,12 @@ This command will print 'Hello, World!' to the console.""" coder.total_cost = 0 coder.cur_messages = [] coder.done_messages = [] + coder.summarizer = MagicMock() + coder.summarizer.too_big.return_value = False coder.cur_messages = [] coder.done_messages = [] + coder.summarizer = MagicMock() + coder.summarizer.too_big.return_value = False # Mock editor_coder creation and execution mock_editor = MagicMock()