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()