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

test: update test assertions to check model objects instead of names

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-12-07 13:38:57 -08:00
parent ccf460c1f7
commit c9c2d5ab6f
2 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/basic/test_history.py
View file

@ -91,9 +91,9 @@ class TestChatSummary(TestCase):
# Check that both models were tried
self.assertEqual(mock_send.call_count, 2)
# Check that the calls were made with the correct model names
self.assertEqual(mock_send.call_args_list[0][0][0], "gpt-4")
self.assertEqual(mock_send.call_args_list[1][0][0], "gpt-3.5-turbo")
# Check that the calls were made with the correct models
self.assertEqual(mock_send.call_args_list[0][0][0], mock_model1)
self.assertEqual(mock_send.call_args_list[1][0][0], mock_model2)
# Check that we got a summary from the second model
self.assertEqual(