Ed_
60396f03f8
Applied 236 return type annotations to functions with no return values across 100+ files (core modules, tests, scripts, simulations). Added Phase 4 to python_style_refactor track for remaining 597 items (untyped params, vars, and functions with return values). Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
17 lines
514 B
Python
17 lines
514 B
Python
import pytest
|
|
from unittest.mock import patch, MagicMock
|
|
import ai_client
|
|
|
|
def test_list_models_gemini_cli() -> None:
|
|
"""
|
|
Verifies that 'ai_client.list_models' correctly returns a list of models
|
|
for the 'gemini_cli' provider.
|
|
"""
|
|
models = ai_client.list_models("gemini_cli")
|
|
assert "gemini-3.1-pro-preview" in models
|
|
assert "gemini-3-flash-preview" in models
|
|
assert "gemini-2.5-pro" in models
|
|
assert "gemini-2.5-flash" in models
|
|
assert "gemini-2.5-flash-lite" in models
|
|
assert len(models) == 5
|