From b35bd06eb8ae2dba900d9fa34e96c088d23d4cdd Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Wed, 2 Oct 2024 11:21:17 -0700 Subject: [PATCH] style: format code in test_models.py --- tests/basic/test_models.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/tests/basic/test_models.py b/tests/basic/test_models.py index a81050963..77ed041b3 100644 --- a/tests/basic/test_models.py +++ b/tests/basic/test_models.py @@ -69,11 +69,15 @@ class TestModels(unittest.TestCase): 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 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] - 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__":