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

chapter5/code-for-math/test_campaign.py

@@ -0,0 +1,74 @@
+from build_aime_2024 import convert
+from demo import campaign_completion, paired_statistics
+
+
+def test_aime_converter_rejects_non_complete_fixture():
+    row = {
+        "id": 1,
+        "problem": "What is 1+1?",
+        "answer": "2",
+        "url": "https://example.test/aime",
+        "year": "2024",
+    }
+    try:
+        convert([row])
+    except ValueError as exc:
+        assert "expected 30" in str(exc)
+    else:
+        raise AssertionError("partial AIME source must not be accepted as the full benchmark")
+
+
+def test_paired_statistics_detects_code_gain():
+    rows = [
+        {
+            "cot_ok": i < 3,
+            "code_ok": i < 10,
+            "used_math_library": i == 0,
+            "tool_calls": 1,
+        }
+        for i in range(10)
+    ]
+    result = paired_statistics(rows)
+    assert result["code_accuracy"] == 1.0
+    assert result["acceptance"]["code_significantly_higher_than_cot"] is True
+
+
+def test_completion_requires_exact_30_task_two_arm_evidence():
+    rows = []
+    for division in ("I", "II"):
+        for number in range(1, 16):
+            rows.append({
+                "id": f"source-row-{division}-{number}",
+                "source": {
+                    "problem_url": (
+                        "https://artofproblemsolving.com/wiki/index.php/"
+                        f"2024_AIME_{division}_Problems/Problem_{number}"
+                    ),
+                },
+                "cot_evidence": {"provider_receipts": [{"response_id": "r"}]},
+                "cot_error": None,
+                "code_evidence": {"provider_receipts": [{"response_id": "r"}]},
+                "code_error": None,
+                "tool_calls": 1,
+            })
+    manifest = {
+        "dataset": "HuggingFaceH4/aime_2024",
+        "revision": "2fe88a2f1091d5048c0f36abc874fb997b3dd99a",
+        "source_sha256": "26139847601a5037c237d5928b195e7260ca8074cf4f264b794af42847f79ccf",
+        "problems": 30,
+        "selection": "all published AIME I and AIME II 2024 problems",
+    }
+    result = campaign_completion(rows, "both", manifest)
+    assert result["status"] == "complete"
+    rows[-1]["tool_calls"] = 0
+    result = campaign_completion(rows, "both", manifest)
+    assert result["status"] == "incomplete"
+    assert result["checks"]["every_code_trajectory_called_real_sandbox"] is False
+
+
+def test_paired_statistics_empty_rows():
+    result = paired_statistics([])
+    assert result["n"] == 0
+    assert result["cot_accuracy"] == 0.0
+    assert result["code_accuracy"] == 0.0
+    assert result["math_library_use_rate"] == 0.0