mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-07-31 19:16:29 +00:00
6b81590c55
Systematic prune per AGENTS.md test policy, one pass over every major test tree (gateway, hermes_cli, tools, agent, run_agent, plugins, cli, cron, tui_gateway, honcho/openviking, root-level): - DELETE: source-reading tests (read_text/getsource on prod files), change-detector tests (exact catalog counts, model-name snapshots, config version literals), mock-echo tests (assert a mock returns what it was told), assertion-free/trivial tests, near-duplicate parametrizations (boundaries + one representative kept), async/sync twin duplicates, cosmetic within-file variations. - KEEP (mandatory): security/redaction/approval guards, message-role alternation invariants, prompt-caching/deterministic-call-id invariants, issue-number regression tests (deduped), E2E tests. - 6 test files deleted outright (script-style/no-assert or fully redundant); conftest.py, fakes/, fixtures/ untouched. - tests/acp/conftest.py added: autouse fixture stubs the live models.dev/GitHub/Copilot/Anthropic inventory fetches that ACP server tests performed on every session create — test_server.py 147s → 3.4s, and the tests are now genuinely hermetic. - Sleep-based slowness shrunk where safe (codex_ttfb_watchdog, compression_concurrent_fork, etc.); no wall-clock assertion tightened. Verification: full hermetic suite via scripts/run_tests.sh — 2439 files, 31,130 tests passed, 0 failed, 0 flaky retries, 315s wall (baseline: 583s wall, 13,564s subprocess CPU).
60 lines
1.4 KiB
Python
60 lines
1.4 KiB
Python
"""Tests for _estimate_chunk_bytes — the cheap per-chunk stream size proxy.
|
|
|
|
Replaced ``len(repr(chunk))`` in the streaming hot loop. Contract:
|
|
- proportional to delta payload length (content, reasoning, tool args);
|
|
- never raises on unknown shapes (returns the framing floor);
|
|
- monotonic accumulation stays positive.
|
|
"""
|
|
|
|
from openai.types.chat import ChatCompletionChunk
|
|
from openai.types.chat.chat_completion_chunk import (
|
|
Choice,
|
|
ChoiceDelta,
|
|
ChoiceDeltaToolCall,
|
|
ChoiceDeltaToolCallFunction,
|
|
)
|
|
|
|
from agent.chat_completion_helpers import _estimate_chunk_bytes
|
|
|
|
|
|
def _chunk(delta):
|
|
return ChatCompletionChunk(
|
|
id="chatcmpl-1",
|
|
created=1730000000,
|
|
model="m",
|
|
object="chat.completion.chunk",
|
|
choices=[Choice(index=0, delta=delta, finish_reason=None)],
|
|
)
|
|
|
|
|
|
|
|
|
|
def test_reasoning_content_counted():
|
|
base = _estimate_chunk_bytes(_chunk(ChoiceDelta()))
|
|
|
|
class Delta:
|
|
content = None
|
|
reasoning_content = "thinking..."
|
|
reasoning = None
|
|
tool_calls = None
|
|
|
|
class Choice0:
|
|
delta = Delta()
|
|
|
|
class Chunk:
|
|
choices = [Choice0()]
|
|
|
|
assert _estimate_chunk_bytes(Chunk()) == base + len("thinking...")
|
|
|
|
|
|
|
|
|
|
def test_unknown_shape_returns_floor_never_raises():
|
|
class Weird:
|
|
pass
|
|
|
|
assert _estimate_chunk_bytes(Weird()) == 40
|
|
assert _estimate_chunk_bytes(None) == 40
|
|
assert _estimate_chunk_bytes(object()) == 40
|
|
|
|
|