style: format code in test_models.py

This commit is contained in:
Paul Gauthier (aider) 2024-10-02 11:19:25 -07:00
parent 1a3e8b3375
commit 1a355cbf74

View file

@ -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__":