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

feat: Add model dumps to test_repo.py

Browse source
This commit is contained in:
Paul Gauthier 2024-08-23 15:17:15 -07:00 • committed by Paul Gauthier (aider)
parent 2906dcb642
commit 100cca5dbf
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/basic/test_repo.py
View file

@ -112,6 +112,8 @@ class TestRepo(unittest.TestCase):
model1 = Model("gpt-3.5-turbo")
model2 = Model("gpt-4")
dump(model1)
dump(model2)
repo = GitRepo(InputOutput(), None, None, models=[model1, model2])
# Call the get_commit_message method with dummy diff and context
@ -123,6 +125,8 @@ class TestRepo(unittest.TestCase):
# Check that simple_send_with_retries was called twice
self.assertEqual(mock_send.call_count, 2)
dump(mock_send.call_args)
# Check that it was called with the correct model names
mock_send.assert_any_call(model1.name, mock_send.call_args[0][1])
mock_send.assert_any_call(model2.name, mock_send.call_args[0][1])