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) 2024-11-26 07:04:21 -08:00
parent e3efab7fbf
commit df1d259e42
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/basic/test_coder.py
View file

@ -884,17 +884,19 @@ This command will print 'Hello, World!' to the console."""
invalid_format = "invalid_format" invalid_format = "invalid_format"
valid_formats = ["diff", "whole", "map"] valid_formats = ["diff", "whole", "map"]
exc = UnknownEditFormat(invalid_format, valid_formats) exc = UnknownEditFormat(invalid_format, valid_formats)
expected_msg = f"Unknown edit format {invalid_format}. Valid formats are: {', '.join(valid_formats)}" expected_msg = (
f"Unknown edit format {invalid_format}. Valid formats are: {', '.join(valid_formats)}"
)
self.assertEqual(str(exc), expected_msg) self.assertEqual(str(exc), expected_msg)
def test_unknown_edit_format_creation(self): def test_unknown_edit_format_creation(self):
# Test that creating a Coder with invalid edit format raises the exception # Test that creating a Coder with invalid edit format raises the exception
io = InputOutput(yes=True) io = InputOutput(yes=True)
invalid_format = "invalid_format" invalid_format = "invalid_format"
with self.assertRaises(UnknownEditFormat) as cm: with self.assertRaises(UnknownEditFormat) as cm:
Coder.create(self.GPT35, invalid_format, io=io) Coder.create(self.GPT35, invalid_format, io=io)
exc = cm.exception exc = cm.exception
self.assertEqual(exc.edit_format, invalid_format) self.assertEqual(exc.edit_format, invalid_format)
self.assertIsInstance(exc.valid_formats, list) self.assertIsInstance(exc.valid_formats, list)