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

chapter8/prompt-distillation/test_create_data_empty.py

@@ -0,0 +1,37 @@
+import asyncio
+
+from create_data import generate_distillation_data
+
+
+def test_generate_empty_input_file(tmp_path):
+    # 空输入文件（0 个句子）：应干净地直接返回，不加载模型、不抛 IndexError/ZeroDivisionError
+    input_file = tmp_path / "empty.txt"
+    input_file.write_text("", encoding="utf-8")
+    output_file = tmp_path / "out.jsonl"
+
+    result = asyncio.run(
+        generate_distillation_data(
+            input_file=str(input_file),
+            output_file=str(output_file),
+            model_name="stub",
+        )
+    )
+    assert result is None
+    assert not output_file.exists()
+
+
+def test_generate_blank_lines_only(tmp_path):
+    # 只有空白行的文件同样视为空
+    input_file = tmp_path / "blank.txt"
+    input_file.write_text("\n   \n\t\n", encoding="utf-8")
+    output_file = tmp_path / "out.jsonl"
+
+    result = asyncio.run(
+        generate_distillation_data(
+            input_file=str(input_file),
+            output_file=str(output_file),
+            model_name="stub",
+        )
+    )
+    assert result is None
+    assert not output_file.exists()