24 lines
810 B
Python
24 lines
810 B
Python
"""嵌入性能基准测试."""
|
|
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)
|