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

feat: Add test for --sonnet --cache-prompts --exit options

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-19 16:05:02 -07:00
parent f9471fc5b6
commit 75a7a0043a
1 changed files with 18 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/basic/test_main.py
View file

@ -533,3 +533,21 @@ class TestMain(TestCase):
self.assertEqual(
call_kwargs.get("refresh"), "files"
) # Check the 'refresh' keyword argument
def test_sonnet_and_cache_prompts_options(self):
with GitTemporaryDirectory():
with patch("aider.coders.Coder.create") as MockCoder:
mock_coder = MagicMock()
MockCoder.return_value = mock_coder
main(
["--sonnet", "--cache-prompts", "--exit", "--yes"],
input=DummyInput(),
output=DummyOutput(),
)
MockCoder.assert_called_once()
_, kwargs = MockCoder.call_args
self.assertEqual(kwargs["main_model"].name, "gpt-4-1106-preview")
self.assertTrue(kwargs["cache_prompts"])
self.assertTrue(mock_coder.add_cache_headers)