18 lines
535 B
Python
18 lines
535 B
Python
import pytest
|
|
from unittest.mock import patch, MagicMock
|
|
import ai_client
|
|
|
|
def test_list_models_gemini_cli():
|
|
"""
|
|
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
|