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

style: format code in test_models.py

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-10-02 11:21:17 -07:00
parent 875fb3a178
commit b35bd06eb8
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/basic/test_models.py
View file

@ -69,11 +69,15 @@ class TestModels(unittest.TestCase):
result = sanity_check_models(mock_io, main_model) result = 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_with(ANY) # Ensure a warning was issued mock_io.tool_warning.assert_called_with(ANY) # Ensure a warning was issued
self.assertEqual(mock_io.tool_warning.call_count, 2) # Expect two warnings self.assertEqual(mock_io.tool_warning.call_count, 2) # Expect two warnings
warning_messages = [call.args[0] for call in mock_io.tool_warning.call_args_list] warning_messages = [call.args[0] for call in mock_io.tool_warning.call_args_list]
self.assertTrue(any("bogus-model" in msg for msg in warning_messages)) # Check that one of the warnings mentions the bogus model self.assertTrue(
any("bogus-model" in msg for msg in warning_messages)
) # Check that one of the warnings mentions the bogus model
if __name__ == "__main__": if __name__ == "__main__":