wehub-resource-sync
75 lines
2.1 KiB
Python
75 lines
2.1 KiB
Python
"""Tests for llms.adapters._content — shared multimodal helpers."""
|
|
|
|
import pytest
|
|
|
|
from omnigent.llms.adapters._content import parse_data_uri
|
|
|
|
|
|
def test_parse_data_uri_png() -> None:
|
|
"""Standard image/png data URI parses correctly."""
|
|
result = parse_data_uri("data:image/png;base64,iVBORw0KGgo=")
|
|
assert result is not None
|
|
assert result.media_type == "image/png"
|
|
assert result.data == "iVBORw0KGgo="
|
|
|
|
|
|
def test_parse_data_uri_pdf() -> None:
|
|
"""Application/pdf data URI parses correctly."""
|
|
result = parse_data_uri("data:application/pdf;base64,JVBERi0=")
|
|
assert result is not None
|
|
assert result.media_type == "application/pdf"
|
|
assert result.data == "JVBERi0="
|
|
|
|
|
|
def test_parse_data_uri_returns_none_for_https() -> None:
|
|
"""External HTTPS URLs return None — not a data URI."""
|
|
assert parse_data_uri("https://example.com/photo.png") is None
|
|
|
|
|
|
def test_parse_data_uri_returns_none_for_http() -> None:
|
|
"""External HTTP URLs return None — not a data URI."""
|
|
assert parse_data_uri("http://example.com/photo.png") is None
|
|
|
|
|
|
def test_parse_data_uri_returns_none_for_missing_base64() -> None:
|
|
"""
|
|
Data URI without ;base64, separator returns None —
|
|
we only support base64-encoded data URIs.
|
|
"""
|
|
assert parse_data_uri("data:text/plain,Hello") is None
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("uri", "expected_type", "expected_data"),
|
|
[
|
|
pytest.param(
|
|
"data:image/jpeg;base64,/9j/4AAQ",
|
|
"image/jpeg",
|
|
"/9j/4AAQ",
|
|
id="jpeg",
|
|
),
|
|
pytest.param(
|
|
"data:image/webp;base64,UklGR",
|
|
"image/webp",
|
|
"UklGR",
|
|
id="webp",
|
|
),
|
|
pytest.param(
|
|
"data:audio/wav;base64,UklF",
|
|
"audio/wav",
|
|
"UklF",
|
|
id="audio",
|
|
),
|
|
],
|
|
)
|
|
def test_parse_data_uri_various_types(
|
|
uri: str,
|
|
expected_type: str,
|
|
expected_data: str,
|
|
) -> None:
|
|
"""Various media types parse correctly."""
|
|
result = parse_data_uri(uri)
|
|
assert result is not None
|
|
assert result.media_type == expected_type
|
|
assert result.data == expected_data
|