Merge pull request #42137 from BerriAI/litellm_migrate_tests_p12

test: migrate phase 12 legacy llm provider tests to tests/unit
This commit is contained in:
yuneng-jiang 2026-09-20 07:49:20 -07:00 • committed by GitHub
commit e0c7a4b19e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
45 changed files with 1 additions and 22 deletions

View file

View file

View file

View file

View file

View file

View file

View file

View file

View file

@ -55,10 +55,6 @@ def _search_kwargs(**overrides):
class TestS3VectorsVectorStoreConfig:
def test_init(self):
config = S3VectorsVectorStoreConfig()
assert config is not None
def test_get_supported_openai_params(self):
config = S3VectorsVectorStoreConfig()
params = config.get_supported_openai_params("test-model")

View file

View file

View file

View file

View file

View file

@ -10,10 +10,7 @@ from unittest.mock import MagicMock
import httpx
import pytest
from litellm.llms.stability.image_generation import (
StabilityImageGenerationConfig,
get_stability_image_generation_config,
)
from litellm.llms.stability.image_generation import StabilityImageGenerationConfig
from litellm.types.llms.stability import (
OPENAI_SIZE_TO_STABILITY_ASPECT_RATIO,
STABILITY_GENERATION_MODELS,
@ -266,20 +263,6 @@ class TestStabilityImageGenerationConfig:
assert "filtered" in str(exc_info.value).lower()
class TestFactoryFunction:
"""Test the factory function"""
def test_get_stability_image_generation_config(self):
"""Test that factory returns correct config type"""
config = get_stability_image_generation_config("stability/sd3")
assert isinstance(config, StabilityImageGenerationConfig)
def test_factory_returns_config_for_any_model(self):
"""Test that factory works for any model name"""
config = get_stability_image_generation_config("stability/custom-model")
assert isinstance(config, StabilityImageGenerationConfig)
class TestOpenAISizeMapping:
"""Test the size to aspect ratio mapping"""

View file

View file