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

test: update sanity check tests to use tool_output instead of tool_warning

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-09-03 17:21:06 -07:00
parent 9e1ca487e2
commit 98835d1f6d
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) sanity_check_model(mock_io, model)
mock_io.tool_warning.assert_called() mock_io.tool_output.assert_called()
calls = mock_io.tool_warning.call_args_list calls = mock_io.tool_output.call_args_list
self.assertIn("- API_KEY1: ✓ Set", str(calls)) self.assertIn("- API_KEY1: ✓ Set", str(calls))
self.assertIn("- API_KEY2: ✓ Set", str(calls)) self.assertIn("- API_KEY2: ✓ Set", str(calls))
@ -57,8 +57,8 @@ class TestModels(unittest.TestCase):
sanity_check_model(mock_io, model) sanity_check_model(mock_io, model)
mock_io.tool_warning.assert_called() mock_io.tool_output.assert_called()
calls = mock_io.tool_warning.call_args_list calls = mock_io.tool_output.call_args_list
self.assertIn("- API_KEY1: ✗ Not set", str(calls)) self.assertIn("- API_KEY1: ✗ Not set", str(calls))
self.assertIn("- API_KEY2: ✗ Not set", str(calls)) self.assertIn("- API_KEY2: ✗ Not set", str(calls))