style: format code with linter

This commit is contained in:
Paul Gauthier (aider) 2024-08-01 15:42:48 -03:00
parent 96940a319a
commit bde6bb5c18

View file

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