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

chapter3/memobase/test_evaluate_generic_empty_query.py

@@ -0,0 +1,38 @@
+"""Regression tests: benchmark evaluation must not raise ZeroDivisionError on
+tasks with an empty/whitespace query (e.g. loaded from tasks.json) that fall
+through to the generic evaluator."""
+import os
+import sys
+
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+
+from locomo_benchmark import BenchmarkTask, LOCOMOBenchmark
+
+LONG_RESPONSE = (
+    "This is a reasonably long agent response. It has several sentences. "
+    "And it keeps going with more detail. Even more detail here."
+)
+
+
+def _suite():
+    return LOCOMOBenchmark.__new__(LOCOMOBenchmark)
+
+
+def test_evaluate_generic_empty_query():
+    task = BenchmarkTask(id="t1", category="custom_category", query="")
+    score = _suite()._evaluate_generic(task, LONG_RESPONSE)
+    assert 0.0 <= score <= 1.0
+
+
+def test_evaluate_response_whitespace_query_unknown_category():
+    task = BenchmarkTask(id="t2", category="custom_category", query="   ")
+    result = _suite().evaluate_response(
+        task, response=LONG_RESPONSE, execution_time=1.0, memory_usage={}
+    )
+    assert 0.0 <= result.score <= 1.0
+
+
+def test_evaluate_generic_normal_query_unchanged():
+    task = BenchmarkTask(id="t3", category="custom_category", query="weather forecast analysis")
+    score = _suite()._evaluate_generic(task, LONG_RESPONSE.replace("agent", "weather"))
+    assert score > 0.0