From 4c35f88ea0ceef293cdf6ef49e52a24bdd862ff2 Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <paul@aider.chat>
Date: Sat, 8 Mar 2025 13:47:31 -0800
Subject: [PATCH] test: add variant flag formats for model switches

---
 tests/basic/test_deprecated.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/tests/basic/test_deprecated.py b/tests/basic/test_deprecated.py
index b1ba3e22d..c82b69037 100644
--- a/tests/basic/test_deprecated.py
+++ b/tests/basic/test_deprecated.py
@@ -31,10 +31,14 @@ class TestDeprecated(TestCase):
             "--sonnet",
             "--haiku",
             "--4",
+            "-4",
             "--4o",
             "--mini",
             "--4-turbo",
+            "--35turbo",
+            "--35-turbo",
             "--3",
+            "-3",
             "--deepseek",
             "--o1-mini",
             "--o1-preview",
@@ -99,10 +103,15 @@ class TestDeprecated(TestCase):
             ("sonnet", "anthropic/claude-3-7-sonnet-20250219"),
             ("haiku", "claude-3-5-haiku-20241022"),
             ("4", "gpt-4-0613"),
+            # Testing the dash variant with underscore in attribute name
+            ("_4", "gpt-4-0613"),
             ("4o", "gpt-4o"),
             ("mini", "gpt-4o-mini"),
             ("4_turbo", "gpt-4-1106-preview"),
             ("35turbo", "gpt-3.5-turbo"),
+            ("35_turbo", "gpt-3.5-turbo"),
+            ("3", "gpt-3.5-turbo"),
+            ("_3", "gpt-3.5-turbo"),
             ("deepseek", "deepseek/deepseek-chat"),
             ("o1_mini", "o1-mini"),
             ("o1_preview", "o1-preview"),