mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-05 02:41:56 +00:00
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:
commit
e0c7a4b19e
45 changed files with 1 additions and 22 deletions
0
tests/unit/llms/openrouter/responses/__init__.py
Normal file
0
tests/unit/llms/openrouter/responses/__init__.py
Normal file
0
tests/unit/llms/parallel_ai/__init__.py
Normal file
0
tests/unit/llms/parallel_ai/__init__.py
Normal file
0
tests/unit/llms/parasail/__init__.py
Normal file
0
tests/unit/llms/parasail/__init__.py
Normal file
0
tests/unit/llms/perplexity/__init__.py
Normal file
0
tests/unit/llms/perplexity/__init__.py
Normal file
0
tests/unit/llms/perplexity/chat/__init__.py
Normal file
0
tests/unit/llms/perplexity/chat/__init__.py
Normal file
0
tests/unit/llms/perplexity/embedding/__init__.py
Normal file
0
tests/unit/llms/perplexity/embedding/__init__.py
Normal file
0
tests/unit/llms/perplexity/responses/__init__.py
Normal file
0
tests/unit/llms/perplexity/responses/__init__.py
Normal file
0
tests/unit/llms/publicai/__init__.py
Normal file
0
tests/unit/llms/publicai/__init__.py
Normal file
0
tests/unit/llms/ragflow/__init__.py
Normal file
0
tests/unit/llms/ragflow/__init__.py
Normal file
0
tests/unit/llms/ragflow/chat/__init__.py
Normal file
0
tests/unit/llms/ragflow/chat/__init__.py
Normal file
0
tests/unit/llms/recraft/__init__.py
Normal file
0
tests/unit/llms/recraft/__init__.py
Normal file
0
tests/unit/llms/recraft/image_edit/__init__.py
Normal file
0
tests/unit/llms/recraft/image_edit/__init__.py
Normal file
0
tests/unit/llms/recraft/image_generation/__init__.py
Normal file
0
tests/unit/llms/recraft/image_generation/__init__.py
Normal file
0
tests/unit/llms/runwayml/__init__.py
Normal file
0
tests/unit/llms/runwayml/__init__.py
Normal file
0
tests/unit/llms/runwayml/videos/__init__.py
Normal file
0
tests/unit/llms/runwayml/videos/__init__.py
Normal file
0
tests/unit/llms/s3_vectors/__init__.py
Normal file
0
tests/unit/llms/s3_vectors/__init__.py
Normal file
0
tests/unit/llms/s3_vectors/vector_stores/__init__.py
Normal file
0
tests/unit/llms/s3_vectors/vector_stores/__init__.py
Normal 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")
|
||||
0
tests/unit/llms/sap/__init__.py
Normal file
0
tests/unit/llms/sap/__init__.py
Normal file
0
tests/unit/llms/scaleway/__init__.py
Normal file
0
tests/unit/llms/scaleway/__init__.py
Normal file
0
tests/unit/llms/snowflake/__init__.py
Normal file
0
tests/unit/llms/snowflake/__init__.py
Normal file
0
tests/unit/llms/soniox/__init__.py
Normal file
0
tests/unit/llms/soniox/__init__.py
Normal file
0
tests/unit/llms/stability/__init__.py
Normal file
0
tests/unit/llms/stability/__init__.py
Normal file
0
tests/unit/llms/stability/image_generation/__init__.py
Normal file
0
tests/unit/llms/stability/image_generation/__init__.py
Normal 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"""
|
||||
|
||||
0
tests/unit/llms/tencent/__init__.py
Normal file
0
tests/unit/llms/tencent/__init__.py
Normal file
0
tests/unit/llms/tencent/chat/__init__.py
Normal file
0
tests/unit/llms/tencent/chat/__init__.py
Normal file
Loading…
Add table
Reference in a new issue