Function assert_round_trip

python/tests/test_integration_codecs.py:189–205

Verify that decode(encode(x)) preserves all fields in the original payload. Messages are compared semantically (key order and None values are normalized). All other keys are compared by exact equality.

(codec, payload)

def assert_round_trip(codec, payload):
    """Verify that decode(encode(x)) preserves all fields in the original payload.

    Messages are compared semantically (key order and None values are normalized).
    All other keys are compared by exact equality.
    """
    original = LLMRequest({}, cast(JsonObject, payload))
    annotated = codec.decode(original)
    result = codec.encode(annotated, original)
    for k, v in payload.items():
        if k == "messages":
            # Semantic comparison: normalize both sides
            orig_msgs = [_normalize_message(m) for m in v]
            result_msgs = [_normalize_message(m) for m in result.content[k]]
            assert result_msgs == orig_msgs, f"Messages differ:\n  orig: {orig_msgs}\n  got:  {result_msgs}"
        else:
            assert result.content[k] == v, f"Key {k!r} changed: {v!r} -> {result.content.get(k)!r}"


# ---------------------------------------------------------------------------