style: format code to improve readability

This commit is contained in:
Paul Gauthier (aider) 2024-08-01 15:37:37 -03:00
parent 4eb8be9d06
commit 82a0e5ec40

View file

@ -22,8 +22,10 @@ class TestScriptingAPI(unittest.TestCase):
# Assertions
self.assertEqual(mock_send.call_count, 2)
mock_send.assert_any_call([{'role': 'user', 'content': 'make a script that prints hello world'}])
mock_send.assert_any_call([{'role': 'user', 'content': 'make it say goodbye'}])
mock_send.assert_any_call(
[{"role": "user", "content": "make a script that prints hello world"}]
)
mock_send.assert_any_call([{"role": "user", "content": "make it say goodbye"}])
self.assertEqual(result1, "Changes applied successfully.")
self.assertEqual(result2, "Changes applied successfully.")
@ -41,7 +43,7 @@ class TestScriptingAPI(unittest.TestCase):
result = coder.run("add a new function")
# Assertions
mock_send.assert_called_once_with([{'role': 'user', 'content': 'add a new function'}])
mock_send.assert_called_once_with([{"role": "user", "content": "add a new function"}])
self.assertEqual(result, "New function added successfully.")
self.assertTrue(io.yes) # Check that 'yes' is set to True