feat: add --yes flag to map tokens tests

This commit is contained in:
Paul Gauthier (aider) 2024-08-06 09:04:41 -03:00
parent 07aa969860
commit ce7cc137fb

View file

@ -379,7 +379,7 @@ class TestMain(TestCase):
with GitTemporaryDirectory(): with GitTemporaryDirectory():
with patch("aider.repomap.RepoMap") as MockRepoMap: with patch("aider.repomap.RepoMap") as MockRepoMap:
main( main(
["--model", "gpt-4", "--map-tokens", "0", "--exit"], ["--model", "gpt-4", "--map-tokens", "0", "--exit", "--yes"],
input=DummyInput(), input=DummyInput(),
output=DummyOutput(), output=DummyOutput(),
) )
@ -389,7 +389,7 @@ class TestMain(TestCase):
with GitTemporaryDirectory(): with GitTemporaryDirectory():
with patch("aider.repomap.RepoMap") as MockRepoMap: with patch("aider.repomap.RepoMap") as MockRepoMap:
main( main(
["--model", "gpt-4", "--map-tokens", "1000", "--exit"], ["--model", "gpt-4", "--map-tokens", "1000", "--exit", "--yes"],
input=DummyInput(), input=DummyInput(),
output=DummyOutput(), output=DummyOutput(),
) )