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

fix: Update test_main.py to use the correct RepoMap import path

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

4
tests/basic/test_main.py
View file

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