From 9121026856fb15bae0776004f74d066c2fa83ee9 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Thu, 5 Dec 2024 06:50:49 -0800 Subject: [PATCH] test: reorganize model test cases for better readability --- tests/basic/test_models.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/tests/basic/test_models.py b/tests/basic/test_models.py index c9c8f5c1b..065687929 100644 --- a/tests/basic/test_models.py +++ b/tests/basic/test_models.py @@ -118,6 +118,11 @@ class TestModels(unittest.TestCase): model = Model("opus") self.assertEqual(model.name, "claude-3-opus-20240229") + # Test non-alias passes through unchanged + model = Model("gpt-4") + self.assertEqual(model.name, "gpt-4") + + def test_o1_use_temp_false(self): # Test GitHub Copilot models model = Model("github/o1-mini") self.assertEqual(model.name, "github/o1-mini") @@ -127,10 +132,6 @@ class TestModels(unittest.TestCase): self.assertEqual(model.name, "github/o1-preview") self.assertEqual(model.use_temperature, 0.0) # Should be deterministic - # Test non-alias passes through unchanged - model = Model("gpt-4") - self.assertEqual(model.name, "gpt-4") - def test_aider_extra_model_settings(self): import tempfile