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

chapter9/prompt-auto-optimization/tests/test_baggage_policy_null.py

@@ -0,0 +1,22 @@
+import json
+
+from airline_env import _run_tool
+
+
+def test_baggage_policy_null_cabin():
+    # 模型显式传 {"cabin": null}：应回退到经济舱默认，而不是 TypeError
+    args = json.loads('{"cabin": null}')
+    result = json.loads(_run_tool("get_baggage_policy", args))
+    assert result["cabin"] == "经济舱"
+    assert result["free_allowance"] == "20kg"
+
+
+def test_baggage_policy_missing_cabin():
+    result = json.loads(_run_tool("get_baggage_policy", {}))
+    assert result["cabin"] == "经济舱"
+    assert result["free_allowance"] == "20kg"
+
+
+def test_baggage_policy_business_cabin():
+    result = json.loads(_run_tool("get_baggage_policy", {"cabin": "商务舱"}))
+    assert result["free_allowance"] == "30kg"