Files
hermes-agent/tests/test_fast_safe_load.py
teknium1 524c38a98a test: purge low-value tests, lane py16 (624 removed)
Change-detectors, tautologies, source-reading tests, redundant duplicates,
mock-echo tests and dead/unrunnable tests. Per-test rationale in the lane
ledger (category + reason for every removal).
2026-09-23 03:15:26 -07:00

50 lines
1.4 KiB
Python

"""Invariants for utils.fast_safe_load.
fast_safe_load is a drop-in for yaml.safe_load that prefers the libyaml
CSafeLoader C extension for speed. These tests assert the behavior contract
(it parses identically to safe_load across input shapes), not a snapshot of
any particular document.
"""
import io
import yaml
from utils import fast_safe_load
_DOCS = [
"", # empty document -> None
"a: 1\nb: two\nc: 3.5\n",
"list: [1, 2, 3]\nnested:\n k: v\n flag: true\n empty: null\n",
"name: skill-x\nmetadata:\n hermes:\n tags: [alpha, beta]\n category: devops\n",
"- one\n- two\n- three\n", # top-level sequence
"scalar string", # bare scalar
]
def test_equivalent_to_safe_load_for_strings():
for doc in _DOCS:
assert fast_safe_load(doc) == yaml.safe_load(doc), repr(doc)
def test_equivalent_to_safe_load_for_file_objects():
for doc in _DOCS:
assert fast_safe_load(io.StringIO(doc)) == yaml.safe_load(io.StringIO(doc)), repr(doc)
def test_rejects_arbitrary_python_objects_like_safe_load():
# Safe loaders must not construct arbitrary Python objects. This tag is
# accepted by the unsafe Loader but rejected by Safe/CSafe loaders.
dangerous = "!!python/object/apply:os.system ['echo pwned']\n"
try:
fast_safe_load(dangerous)
raised = False
except yaml.YAMLError:
raised = True
assert raised, "fast_safe_load must reject python/object tags like safe_load"