ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter10/generative-agents/tests/test_validate_campaign.py

@@ -0,0 +1,41 @@
+from __future__ import annotations
+
+from validate_campaign import (
+    canonical_provider_receipt,
+    compatibility_correction_valid,
+    positive_provider_usage,
+)
+
+
+def test_positive_provider_usage_ignores_nested_token_details():
+    row = {
+        "response": {
+            "usage": {
+                "prompt_tokens": 49,
+                "prompt_tokens_details": {"cached_tokens": 0},
+                "total_tokens": 69,
+            }
+        }
+    }
+    assert positive_provider_usage(row) is True
+
+
+def test_compatibility_correction_must_resolve_to_accessible_arena():
+    row = {
+        "kind": "action_arena_compatibility_correction",
+        "raw_output": "{Tom and Jane Moreno's bedroom",
+        "normalized_output": "Tom and Jane Moreno's bedroom",
+        "accessible_arenas": ["common room", "Tom and Jane Moreno's bedroom"],
+        "reason": "stripped_response_wrappers",
+        "fallback": False,
+    }
+    assert compatibility_correction_valid(row) is True
+    row["normalized_output"] = "private vault"
+    assert compatibility_correction_valid(row) is False
+
+
+def test_failed_compressed_receipts_are_not_canonical(tmp_path):
+    assert canonical_provider_receipt(tmp_path / "steps_00000_00360.jsonl.gz")
+    assert not canonical_provider_receipt(
+        tmp_path / "steps_00000_00360.failed-123.jsonl.gz"
+    )