"""嵌入性能基准测试."""
import pytest


def test_bench_embed_single(benchmark, benchmark_embedder):
    """单文本嵌入耗时."""
    text = "这是一段测试文本，用于测量嵌入速度。"
    benchmark(benchmark_embedder.embed, [text])


def test_bench_embed_batch_32(benchmark, benchmark_embedder):
    """批量 32 文本嵌入耗时."""
    texts = [f"测试文本第{i}条，模拟真实文档内容。" for i in range(32)]
    benchmark(benchmark_embedder.embed, texts)


def test_bench_embed_batch_100(benchmark, benchmark_embedder):
    """批量 100 文本嵌入耗时（GPU 优势显著）."""
    texts = [
        f"测试文本第{i}条。Python 通用编程语言，用于数据科学和 AI。"
        for i in range(100)
    ]
    benchmark(benchmark_embedder.embed, texts)