feat: Add test case for --4o and --cache options

2025-05-20 12:24:59 +00:00 · 2024-08-19 16:07:13 -07:00 · 2024-08-19 16:07:13 -07:00 · 621f91cfee
commit 621f91cfee
parent d71ea571e4
1 changed files with 12 additions and 0 deletions
--- a/tests/basic/test_main.py
+++ b/tests/basic/test_main.py
@ -544,3 +544,15 @@ class TestMain(TestCase):
            )

            self.assertTrue(coder.add_cache_headers)
+
+    def test_4o_and_cache_options(self):
+        with GitTemporaryDirectory():
+            coder = main(
+                ["--4o", "--cache", "--exit", "--yes"],
+                input=DummyInput(),
+                output=DummyOutput(),
+                return_coder=True,
+            )
+
+            self.assertFalse(coder.add_cache_headers)
+            self.assertEqual(coder.main_model.name, "gpt-4-1106-preview")