97e91a83f3
Ruff / Ruff (push) Has been cancelled
Test / Core Tests (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.10) (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.11) (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.12) (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.13) (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.9) (push) Has been cancelled
Test / Full Coverage (Python 3.11) (push) Has been cancelled
Test / Core Provider Tests (OpenAI) (push) Has been cancelled
Test / Core Provider Tests (Anthropic) (push) Has been cancelled
Test / Core Provider Tests (Google) (push) Has been cancelled
Test / Core Provider Tests (Other) (push) Has been cancelled
Test / Anthropic Tests (push) Has been cancelled
Test / Gemini Tests (push) Has been cancelled
Test / Google GenAI Tests (push) Has been cancelled
Test / Vertex AI Tests (push) Has been cancelled
Test / OpenAI Tests (push) Has been cancelled
Test / Writer Tests (push) Has been cancelled
Test / Auto Client Tests (push) Has been cancelled
ty / type-check (push) Has been cancelled
27 lines
708 B
Python
27 lines
708 B
Python
from __future__ import annotations
|
|
import pytest
|
|
from instructor.v2.providers.bedrock.handlers import _normalize_bedrock_image_format
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"inp,expected",
|
|
[
|
|
("image/jpeg", "jpeg"),
|
|
("image/jpg", "jpeg"),
|
|
("jpg", "jpeg"),
|
|
("jpeg", "jpeg"),
|
|
("image/pjpeg", "jpeg"),
|
|
("image/png", "png"),
|
|
("png", "png"),
|
|
("image/gif", "gif"),
|
|
("gif", "gif"),
|
|
("image/webp", "webp"),
|
|
("webp", "webp"),
|
|
("", "jpeg"),
|
|
(None, "jpeg"),
|
|
("image/whatever", "jpeg"),
|
|
],
|
|
)
|
|
def test_normalize_bedrock_image_format(inp, expected):
|
|
assert _normalize_bedrock_image_format(inp) == expected
|