dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-20 12:24:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

style: Fix linting issues in test_coder.py

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-02-07 08:39:39 -08:00
parent 52bc51a197
commit af48c46c30
1 changed files with 14 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

28
tests/basic/test_coder.py
View file

@ -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)