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

feat: Add test for main() with --sonnet and --cache-prompts options

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-19 16:06:34 -07:00
parent 1ea18b83b9
commit 3424cda63c
1 changed files with 12 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/basic/test_main.py
View file

@ -533,3 +533,15 @@ class TestMain(TestCase):
self.assertEqual( self.assertEqual(
call_kwargs.get("refresh"), "files" call_kwargs.get("refresh"), "files"
) # Check the 'refresh' keyword argument ) # Check the 'refresh' keyword argument
def test_sonnet_and_cache_prompts_options(self):
with GitTemporaryDirectory():
coder = main(
["--sonnet", "--cache-prompts", "--exit", "--yes"],
input=DummyInput(),
output=DummyOutput(),
return_coder=True,
)
self.assertTrue(coder.add_cache_headers)
self.assertEqual(coder.main_model.name, "gpt-4-1106-preview")