style: Format code with consistent indentation and line breaks

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

View file

@ -378,11 +378,19 @@ class TestMain(TestCase):
def test_map_tokens_option(self):
with GitTemporaryDirectory():
with patch("aider.repomap.RepoMap") as MockRepoMap:
main(["--model", "gpt-4", "--map-tokens", "0", "--exit"], input=DummyInput(), output=DummyOutput())
main(
["--model", "gpt-4", "--map-tokens", "0", "--exit"],
input=DummyInput(),
output=DummyOutput(),
)
MockRepoMap.assert_not_called()
def test_map_tokens_option_with_non_zero_value(self):
with GitTemporaryDirectory():
with patch("aider.repomap.RepoMap") as MockRepoMap:
main(["--model", "gpt-4", "--map-tokens", "1000", "--exit"], input=DummyInput(), output=DummyOutput())
main(
["--model", "gpt-4", "--map-tokens", "1000", "--exit"],
input=DummyInput(),
output=DummyOutput(),
)
MockRepoMap.assert_called_once()