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

fix: Resolve issues in test_main.py

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-06 09:07:40 -03:00
parent ec7a212b60
commit d34da62b12
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/basic/test_main.py
View file

@ -377,7 +377,7 @@ class TestMain(TestCase):
def test_map_tokens_option(self):
with GitTemporaryDirectory():
with patch("aider.base_coder.RepoMap") as MockRepoMap:
with patch("aider.repomap.RepoMap") as MockRepoMap:
main(
["--model", "gpt-4", "--map-tokens", "0", "--exit", "--yes"],
input=DummyInput(),
@ -387,7 +387,8 @@ class TestMain(TestCase):
def test_map_tokens_option_with_non_zero_value(self):
with GitTemporaryDirectory():
with patch("aider.coders.base_coder.RepoMap") as MockRepoMap:
with patch("aider.repomap.RepoMap") as MockRepoMap:
MockRepoMap.return_value.max_map_tokens = 1000
main(
["--model", "gpt-4", "--map-tokens", "1000", "--exit", "--yes"],
input=DummyInput(),