mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-06 02:48:13 +00:00
aecc08050f
Type aliases carry TypeAlias, cast sites carry cast-ok on the call line, dict fallbacks use frozen constants, and nested envelope seeding replaces a None metadata value instead of returning it (legacy behaviour the setdefault form missed). Fixes the PT012 shape in test_bindings and formats ocr/main.
49 lines
1.8 KiB
Python
49 lines
1.8 KiB
Python
from types import SimpleNamespace
|
|
from typing import Final
|
|
|
|
import pytest
|
|
|
|
from litellm.rust_bridge import bindings
|
|
|
|
|
|
def test_binding_distinguishes_disable_from_reset(monkeypatch) -> None:
|
|
native = SimpleNamespace(route=lambda: "native")
|
|
monkeypatch.setattr(bindings, "get_native_bridge", lambda: native)
|
|
binding: bindings.NativeBinding[object] = bindings.NativeBinding("route", validate=lambda value: value)
|
|
|
|
assert binding.load() is native.route
|
|
|
|
binding.override(None)
|
|
assert binding.load() is None
|
|
|
|
replacement = object()
|
|
binding.override(replacement)
|
|
assert binding.load() is replacement
|
|
|
|
binding.reset()
|
|
assert binding.load() is native.route
|
|
|
|
|
|
@pytest.mark.parametrize(("value", "expected"), ((3, 3), ("invalid", None), (None, None)))
|
|
def test_binding_validates_native_attribute(
|
|
monkeypatch: pytest.MonkeyPatch, value: object, expected: int | None
|
|
) -> None:
|
|
native: Final = SimpleNamespace(route=value)
|
|
monkeypatch.setattr(bindings, "get_native_bridge", lambda: native)
|
|
binding: Final = bindings.NativeBinding("route", validate=lambda item: item if isinstance(item, int) else None)
|
|
|
|
assert binding.load() == expected
|
|
|
|
|
|
@pytest.mark.parametrize("available", [False, True])
|
|
def test_decline_accessor_only_catches_admission_declines(monkeypatch: pytest.MonkeyPatch, available: bool) -> None:
|
|
class Declined(Exception):
|
|
pass
|
|
|
|
class Upstream(Exception):
|
|
pass
|
|
|
|
native: Final = SimpleNamespace(RustBridgeDeclined=Declined, RustUpstreamError=Upstream) if available else None
|
|
monkeypatch.setattr(bindings, "get_native_bridge", lambda: native)
|
|
assert bindings.native_decline_types() == ((Declined,) if available else ())
|
|
assert not isinstance(Upstream("already dispatched"), bindings.native_decline_types())
|