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

test: update sanity_check_model tests to use tool_warning

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-09-03 13:50:10 -07:00
parent a6892c0421
commit d8c78cf8cd
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/basic/test_models.py
View file

@ -40,8 +40,8 @@ class TestModels(unittest.TestCase):
sanity_check_model(mock_io, model)
mock_io.tool_error.assert_called()
calls = mock_io.tool_error.call_args_list
mock_io.tool_warning.assert_called()
calls = mock_io.tool_warning.call_args_list
self.assertIn("- API_KEY1: ✓ Set", str(calls))
self.assertIn("- API_KEY2: ✓ Set", str(calls))
@ -57,8 +57,8 @@ class TestModels(unittest.TestCase):
sanity_check_model(mock_io, model)
mock_io.tool_error.assert_called()
calls = mock_io.tool_error.call_args_list
mock_io.tool_warning.assert_called()
calls = mock_io.tool_warning.call_args_list
self.assertIn("- API_KEY1: ✗ Not set", str(calls))
self.assertIn("- API_KEY2: ✗ Not set", str(calls))