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

fix: Update test assertions for API key checks

Browse source
This commit is contained in:
Paul Gauthier 2024-09-23 11:34:49 -07:00 • committed by Paul Gauthier (aider)
parent d518493bb9
commit 924eeb43de
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

@ -42,8 +42,8 @@ class TestModels(unittest.TestCase):
mock_io.tool_output.assert_called()
calls = mock_io.tool_output.call_args_list
self.assertIn("- API_KEY1: ✓ Set", str(calls))
self.assertIn("- API_KEY2: ✓ Set", str(calls))
self.assertIn("- API_KEY1: Set", str(calls))
self.assertIn("- API_KEY2: Set", str(calls))
@patch("os.environ")
def test_sanity_check_model_not_set(self, mock_environ):
@ -59,8 +59,8 @@ class TestModels(unittest.TestCase):
mock_io.tool_output.assert_called()
calls = mock_io.tool_output.call_args_list
self.assertIn("- API_KEY1: ✗ Not set", str(calls))
self.assertIn("- API_KEY2: ✗ Not set", str(calls))
self.assertIn("- API_KEY1: Not set", str(calls))
self.assertIn("- API_KEY2: Not set", str(calls))
if __name__ == "__main__":