From af48c46c30f33870d71afa35160769cf3271230e Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Fri, 7 Feb 2025 08:39:39 -0800 Subject: [PATCH] style: Fix linting issues in test_coder.py --- tests/basic/test_coder.py | 28 ++++++++++++++-------------- 1 file changed, 14 insertions(+), 14 deletions(-) diff --git a/tests/basic/test_coder.py b/tests/basic/test_coder.py index e0488b8c2..4e391ddf1 100644 --- a/tests/basic/test_coder.py +++ b/tests/basic/test_coder.py @@ -909,17 +909,17 @@ This command will print 'Hello, World!' to the console.""" # Test that system_prompt_prefix is properly set and used io = InputOutput(yes=True) test_prefix = "Test prefix. " - + # Create a model with system_prompt_prefix model = Model("gpt-3.5-turbo") model.system_prompt_prefix = test_prefix - + coder = Coder.create(model, None, io=io) - + # Get the formatted messages chunks = coder.format_messages() messages = chunks.all_messages() - + # Check if the system message contains our prefix system_message = next(msg for msg in messages if msg["role"] == "system") self.assertTrue(system_message["content"].startswith(test_prefix)) @@ -927,42 +927,42 @@ This command will print 'Hello, World!' to the console.""" def test_system_prompt_prefix_none(self): # Test behavior when system_prompt_prefix is None io = InputOutput(yes=True) - + # Create a model without system_prompt_prefix model = Model("gpt-3.5-turbo") model.system_prompt_prefix = None - + coder = Coder.create(model, None, io=io) - + # Get the formatted messages chunks = coder.format_messages() messages = chunks.all_messages() - + # Get the system message system_message = next(msg for msg in messages if msg["role"] == "system") original_content = coder.fmt_system_prompt(coder.gpt_prompts.main_system) - + # Check that the content matches the original prompt without prefix self.assertEqual(system_message["content"], original_content) def test_system_prompt_prefix_empty(self): # Test behavior when system_prompt_prefix is empty string io = InputOutput(yes=True) - + # Create a model with empty system_prompt_prefix model = Model("gpt-3.5-turbo") model.system_prompt_prefix = "" - + coder = Coder.create(model, None, io=io) - + # Get the formatted messages chunks = coder.format_messages() messages = chunks.all_messages() - + # Get the system message system_message = next(msg for msg in messages if msg["role"] == "system") original_content = coder.fmt_system_prompt(coder.gpt_prompts.main_system) - + # Check that the content matches the original prompt without prefix self.assertEqual(system_message["content"], original_content)