dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-21 21:04:59 +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_repo.py
View file

@ -125,9 +125,9 @@ class TestRepo(unittest.TestCase):
# Check that simple_send_with_retries was called twice
self.assertEqual(mock_send.call_count, 2)
# Check that it was called with the correct model names
self.assertEqual(mock_send.call_args_list[0][0][0], model1.name)
self.assertEqual(mock_send.call_args_list[1][0][0], model2.name)
# Check that it was called with the correct models
self.assertEqual(mock_send.call_args_list[0][0][0], model1)
self.assertEqual(mock_send.call_args_list[1][0][0], model2)
# Check that the content of the messages is the same for both calls
self.assertEqual(mock_send.call_args_list[0][0][1], mock_send.call_args_list[1][0][1])