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

chapter8/prompt-distillation/test_evaluate_empty_parse_rate.py

@@ -0,0 +1,25 @@
+"""Regression: empty test set must not ZeroDivisionError on parse-rate summary."""
+import sys
+import types
+
+
+def _stub_evaluate_deps() -> None:
+    for name in ["torch", "numpy", "transformers", "peft", "tqdm"]:
+        sys.modules.setdefault(name, types.ModuleType(name))
+    sys.modules["transformers"].AutoTokenizer = object
+    sys.modules["transformers"].AutoModelForCausalLM = object
+    sys.modules["peft"].PeftModel = object
+    sys.modules["tqdm"].tqdm = lambda x, **k: x
+
+
+_stub_evaluate_deps()
+
+from evaluate import compute_parse_rate  # noqa: E402
+
+
+def test_compute_parse_rate_empty_total_is_zero():
+    assert compute_parse_rate(0, 0) == 0.0
+
+
+def test_compute_parse_rate_nonzero_total():
+    assert compute_parse_rate(8, 10) == 0.8