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

test: update config file model and token values in test cases

Browse source
This commit is contained in:
sltptr 2025-05-16 18:54:09 -07:00 • committed by sltptr (aider)
parent 7712145543
commit d8650c3131
1 changed files with 8 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

16
tests/basic/test_main.py
View file

@ -432,9 +432,9 @@ class TestMain(TestCase):
git_config_yml.write_text("model: gpt-4\nmap-tokens: 2048\n") git_config_yml.write_text("model: gpt-4\nmap-tokens: 2048\n")
home_config_yml.write_text("model: gpt-3.5-turbo\nmap-tokens: 1024\n") home_config_yml.write_text("model: gpt-3.5-turbo\nmap-tokens: 1024\n")
named_config.write_text("model: gpt-4-1106-preview\nmap-tokens: 8192\n") named_config.write_text("model: gpt-4-1106-preview\nmap-tokens: 8192\n")
cwd_config_yaml.write_text("model: gpt-4-32k-yaml\nmap-tokens: 9999\n") cwd_config_yaml.write_text("model: gpt-4-32k\nmap-tokens: 4096\n")
git_config_yaml.write_text("model: gpt-4-yaml\nmap-tokens: 8888\n") git_config_yaml.write_text("model: gpt-4\nmap-tokens: 2048\n")
home_config_yaml.write_text("model: gpt-3.5-turbo-yaml\nmap-tokens: 7777\n") home_config_yaml.write_text("model: gpt-3.5-turbo\nmap-tokens: 1024\n")
with ( with (
patch("pathlib.Path.home", return_value=fake_home), patch("pathlib.Path.home", return_value=fake_home),
@ -475,26 +475,26 @@ class TestMain(TestCase):
home_config_yml.unlink(missing_ok=True) home_config_yml.unlink(missing_ok=True)
main(["--yes", "--exit"], input=DummyInput(), output=DummyOutput()) main(["--yes", "--exit"], input=DummyInput(), output=DummyOutput())
_, kwargs = MockCoder.call_args _, kwargs = MockCoder.call_args
self.assertEqual(kwargs["main_model"].name, "gpt-4-32k-yaml") self.assertEqual(kwargs["main_model"].name, "gpt-4-32k")
self.assertEqual(kwargs["map_tokens"], 9999) self.assertEqual(kwargs["map_tokens"], 4096)
# Test loading from .yaml in git root # Test loading from .yaml in git root
cwd_config_yaml.unlink() cwd_config_yaml.unlink()
main(["--yes", "--exit"], input=DummyInput(), output=DummyOutput()) main(["--yes", "--exit"], input=DummyInput(), output=DummyOutput())
_, kwargs = MockCoder.call_args _, kwargs = MockCoder.call_args
self.assertEqual(kwargs["main_model"].name, "gpt-4-yaml") self.assertEqual(kwargs["main_model"].name, "gpt-4-yaml")
self.assertEqual(kwargs["map_tokens"], 8888) self.assertEqual(kwargs["map_tokens"], 2048)
# Test loading from .yaml in home # Test loading from .yaml in home
git_config_yaml.unlink() git_config_yaml.unlink()
main(["--yes", "--exit"], input=DummyInput(), output=DummyOutput()) main(["--yes", "--exit"], input=DummyInput(), output=DummyOutput())
_, kwargs = MockCoder.call_args _, kwargs = MockCoder.call_args
self.assertEqual(kwargs["main_model"].name, "gpt-3.5-turbo-yaml") self.assertEqual(kwargs["main_model"].name, "gpt-3.5-turbo-yaml")
self.assertEqual(kwargs["map_tokens"], 7777) self.assertEqual(kwargs["map_tokens"], 1024)
# Test both .yml and .yaml present, .yml preferred and warning printed # Test both .yml and .yaml present, .yml preferred and warning printed
home_config_yml.write_text("model: gpt-3.5-turbo\nmap-tokens: 1024\n") home_config_yml.write_text("model: gpt-3.5-turbo\nmap-tokens: 1024\n")
home_config_yaml.write_text("model: gpt-3.5-turbo-yaml\nmap-tokens: 7777\n") home_config_yaml.write_text("model: gpt-3.5-turbo\nmap-tokens: 1024\n")
with patch("builtins.print") as mock_print: with patch("builtins.print") as mock_print:
main(["--yes", "--exit"], input=DummyInput(), output=DummyOutput()) main(["--yes", "--exit"], input=DummyInput(), output=DummyOutput())
_, kwargs = MockCoder.call_args _, kwargs = MockCoder.call_args