From 476a0ad6adfda779d68d4627f6756c4c5ba2f0b8 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Fri, 31 Jan 2025 13:13:05 -0800 Subject: [PATCH] test: Add test for --reasoning-effort CLI option --- tests/basic/test_main.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/basic/test_main.py b/tests/basic/test_main.py index 14547bb49..649b6648d 100644 --- a/tests/basic/test_main.py +++ b/tests/basic/test_main.py @@ -824,3 +824,7 @@ class TestMain(TestCase): self.fail(f"main() raised an unexpected exception: {e}") self.assertIsNone(result, "main() should return None when called with --exit") + + def test_reasoning_effort_option(self): + coder = main(["--reasoning-effort", "3", "--yes", "--exit"], input=DummyInput(), output=DummyOutput(), return_coder=True) + self.assertEqual(coder.main_model.extra_params.get("extra_body", {}).get("reasoning_effort"), "3")