From 45f6f88a4664344ff49348c0f7d7273eae226e8b Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Tue, 6 Aug 2024 09:04:01 -0300 Subject: [PATCH] feat: Add tests for --map-tokens option --- tests/basic/test_main.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/tests/basic/test_main.py b/tests/basic/test_main.py index fa7a3a394..6adecb418 100644 --- a/tests/basic/test_main.py +++ b/tests/basic/test_main.py @@ -374,3 +374,15 @@ class TestMain(TestCase): self.assertIn("dark_mode", relevant_output) self.assertRegex(relevant_output, r"AIDER_DARK_MODE:\s+on") self.assertRegex(relevant_output, r"dark_mode:\s+True") + + 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()) + 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()) + MockRepoMap.assert_called_once()