ai-agent-book/chapter8/prompt-distillation/test_create_data_empty.py

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()