Gerald LonlasandClaude Sonnet 4.6
617100dbe6
- coerce_seed("") returns None, not ValueError; move "" out of the
invalid-values parametrize and give it its own explicit assertion
- strip_think_content tests used \\n (literal backslash-n) instead of
real newlines; .strip() cannot remove backslash or 'n' so the dangling-
tag test always failed. Replace with actual \n to match real LLM output.
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
67 lines
2.1 KiB
Python
67 lines
2.1 KiB
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
import client
|
|
|
|
|
|
def test_normalize_server_url_trims_and_strips_v1() -> None:
|
|
assert client.normalize_server_url(" http://10.168.168.7:1234/v1/ ") == "http://10.168.168.7:1234"
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"value,expected",
|
|
[
|
|
(None, None),
|
|
(42, 42),
|
|
(42.9, 42),
|
|
("17", 17),
|
|
(b"99", 99),
|
|
([123], 123),
|
|
((456,), 456),
|
|
],
|
|
)
|
|
def test_coerce_seed_supported_values(value, expected) -> None:
|
|
assert client.coerce_seed(value) == expected
|
|
|
|
|
|
@pytest.mark.parametrize("value", [[], [1, 2], "abc", float("nan")])
|
|
def test_coerce_seed_invalid_values(value) -> None:
|
|
with pytest.raises(ValueError):
|
|
client.coerce_seed(value)
|
|
|
|
|
|
def test_coerce_seed_empty_string_returns_none() -> None:
|
|
# Empty string is treated as "no seed provided" → None → random seed downstream.
|
|
assert client.coerce_seed("") is None
|
|
|
|
|
|
def test_resolve_request_seed_negative_one_generates_positive_seed() -> None:
|
|
resolved = client.resolve_request_seed(-1)
|
|
assert isinstance(resolved, int)
|
|
assert resolved > 0
|
|
|
|
|
|
def test_build_responses_input_text_shape() -> None:
|
|
payload = client.build_responses_input_text("hello")
|
|
assert payload[0]["role"] == "user"
|
|
assert payload[0]["content"][0]["type"] == "input_text"
|
|
assert payload[0]["content"][0]["text"] == "hello"
|
|
|
|
|
|
def test_build_chat_messages_handles_none_prompts() -> None:
|
|
messages = client.build_chat_messages(None, None)
|
|
assert messages == [{"role": "user", "content": ""}]
|
|
|
|
|
|
def test_strip_think_content_removes_reasoning_blocks() -> None:
|
|
# Use real newlines — as actual LLM output would contain.
|
|
text = "Final answer\n<think>internal chain of thought</think>\nVisible output"
|
|
assert client.strip_think_content(text) == "Final answer\n\nVisible output"
|
|
|
|
|
|
def test_strip_think_content_removes_dangling_start_tag() -> None:
|
|
# strip() removes the trailing real newline left after the dangling block is excised.
|
|
text = "Visible output\n<think>unfinished"
|
|
assert client.strip_think_content(text) == "Visible output"
|