dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-01 10:14:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

test: Update OpenRouter default model expectations

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-05-30 15:07:29 -07:00
parent c24798c44f
commit 52510c7da5
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/basic/test_onboarding.py
View file

@ -94,7 +94,7 @@ class TestOnboarding(unittest.TestCase):
def test_try_select_default_model_openrouter_free(self, mock_check_tier):
"""Test OpenRouter free model selection."""
self.assertEqual(
try_to_select_default_model(), "openrouter/google/gemini-2.5-pro-exp-03-25:free"
try_to_select_default_model(), "openrouter/deepseek/deepseek-r1:free"
)
mock_check_tier.assert_called_once_with("or_key")
@ -102,7 +102,7 @@ class TestOnboarding(unittest.TestCase):
@patch.dict(os.environ, {"OPENROUTER_API_KEY": "or_key"}, clear=True)
def test_try_select_default_model_openrouter_paid(self, mock_check_tier):
"""Test OpenRouter paid model selection."""
self.assertEqual(try_to_select_default_model(), "openrouter/anthropic/claude-3.7-sonnet")
self.assertEqual(try_to_select_default_model(), "openrouter/anthropic/claude-sonnet-4")
mock_check_tier.assert_called_once_with("or_key")
@patch("aider.onboarding.check_openrouter_tier")
@ -146,7 +146,7 @@ class TestOnboarding(unittest.TestCase):
)
def test_try_select_default_model_priority_openrouter(self, mock_check_tier):
"""Test OpenRouter key takes priority."""
self.assertEqual(try_to_select_default_model(), "openrouter/anthropic/claude-3.7-sonnet")
self.assertEqual(try_to_select_default_model(), "openrouter/anthropic/claude-sonnet-4")
mock_check_tier.assert_called_once_with("or_key")
@patch("aider.onboarding.check_openrouter_tier")
@ -346,7 +346,7 @@ class TestOnboarding(unittest.TestCase):
@patch(
"aider.onboarding.try_to_select_default_model",
side_effect=[None, "openrouter/google/gemini-2.5-pro-exp-03-25:free"],
side_effect=[None, "openrouter/deepseek/deepseek-r1:free"],
) # Fails first, succeeds after oauth
@patch(
"aider.onboarding.offer_openrouter_oauth", return_value=True
@ -360,7 +360,7 @@ class TestOnboarding(unittest.TestCase):
selected_model = select_default_model(args, io_mock, analytics_mock)
self.assertEqual(selected_model, "openrouter/google/gemini-2.5-pro-exp-03-25:free")
self.assertEqual(selected_model, "openrouter/deepseek/deepseek-r1:free")
self.assertEqual(mock_try_select.call_count, 2) # Called before and after oauth
mock_offer_oauth.assert_called_once_with(io_mock, analytics_mock)
# Only one warning is expected: "No LLM model..."