mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-24 06:15:00 +00:00
style: format code in test_models.py
This commit is contained in:
parent
1a3e8b3375
commit
1a355cbf74
1 changed files with 6 additions and 2 deletions
|
@ -69,10 +69,14 @@ class TestModels(unittest.TestCase):
|
||||||
|
|
||||||
result = models.sanity_check_models(mock_io, main_model)
|
result = models.sanity_check_models(mock_io, main_model)
|
||||||
|
|
||||||
self.assertTrue(result) # Should return True because there's a problem with the editor model
|
self.assertTrue(
|
||||||
|
result
|
||||||
|
) # Should return True because there's a problem with the editor model
|
||||||
mock_io.tool_warning.assert_called_once() # Ensure a warning was issued
|
mock_io.tool_warning.assert_called_once() # Ensure a warning was issued
|
||||||
warning_message = mock_io.tool_warning.call_args[0][0]
|
warning_message = mock_io.tool_warning.call_args[0][0]
|
||||||
self.assertIn("bogus-model", warning_message) # Check that the warning mentions the bogus model
|
self.assertIn(
|
||||||
|
"bogus-model", warning_message
|
||||||
|
) # Check that the warning mentions the bogus model
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue