feat: 第三优先级完善 — CI/CD、PyPI、pre-commit、CHANGELOG、贡献指南、基准测试、ADR
CI / Test (Python 3.13) (push) Has been cancelled
CI / Test (Python 3.13) (push) Has been cancelled
This commit is contained in:
@@ -0,0 +1,23 @@
|
||||
"""嵌入性能基准测试."""
|
||||
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)
|
||||
Reference in New Issue
Block a user