50 lines
1.8 KiB
Python
50 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
import unittest
|
|
|
|
from chatmock.fast_mode import parse_optional_bool, resolve_service_tier, supports_priority_service_tier
|
|
|
|
|
|
class FastModeTests(unittest.TestCase):
|
|
def test_parse_optional_bool(self) -> None:
|
|
self.assertTrue(parse_optional_bool(True))
|
|
self.assertTrue(parse_optional_bool("true"))
|
|
self.assertFalse(parse_optional_bool(False))
|
|
self.assertFalse(parse_optional_bool("off"))
|
|
self.assertIsNone(parse_optional_bool("maybe"))
|
|
|
|
def test_priority_allowlist_uses_normalized_model_ids(self) -> None:
|
|
self.assertTrue(supports_priority_service_tier("gpt5.4"))
|
|
self.assertFalse(supports_priority_service_tier("gpt-5.3-codex"))
|
|
|
|
def test_explicit_fast_mode_true_errors_for_unsupported_model(self) -> None:
|
|
resolution = resolve_service_tier(
|
|
"gpt-5.3-codex",
|
|
request_fast_mode=True,
|
|
server_fast_mode=False,
|
|
)
|
|
self.assertIsNone(resolution.service_tier)
|
|
self.assertIsNotNone(resolution.error_message)
|
|
|
|
def test_server_default_fast_mode_falls_back_on_unsupported_model(self) -> None:
|
|
resolution = resolve_service_tier(
|
|
"gpt-5.3-codex",
|
|
server_fast_mode=True,
|
|
)
|
|
self.assertIsNone(resolution.service_tier)
|
|
self.assertIsNone(resolution.error_message)
|
|
self.assertIsNotNone(resolution.warning_message)
|
|
|
|
def test_request_fast_mode_false_overrides_server_default(self) -> None:
|
|
resolution = resolve_service_tier(
|
|
"gpt-5.4",
|
|
request_fast_mode=False,
|
|
server_fast_mode=True,
|
|
)
|
|
self.assertIsNone(resolution.service_tier)
|
|
self.assertIsNone(resolution.error_message)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|