mirror of
https://github.com/rennf93/roboco.git
synced 2026-08-03 07:23:24 +02:00
25 lines
761 B
Python
25 lines
761 B
Python
"""Coverage for roboco.api.schemas.provider helpers."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from types import SimpleNamespace
|
|
from uuid import uuid4
|
|
|
|
from roboco.api.schemas.provider import assignment_to_response
|
|
from roboco.models.base import AssignmentScope, ModelProvider
|
|
|
|
|
|
def test_assignment_to_response_round_trip() -> None:
|
|
provider = SimpleNamespace(type=ModelProvider.ANTHROPIC)
|
|
row = SimpleNamespace(
|
|
id=uuid4(),
|
|
scope=AssignmentScope.GLOBAL,
|
|
scope_value=None,
|
|
provider=provider,
|
|
model_name="opus",
|
|
)
|
|
response = assignment_to_response(row)
|
|
assert response.scope == AssignmentScope.GLOBAL
|
|
assert response.provider_type == ModelProvider.ANTHROPIC
|
|
assert response.model_name == "opus"
|