"""检索性能基准测试."""
import pytest


def test_bench_search_top5(benchmark, benchmark_searcher):
    """top_k=5 检索耗时."""
    benchmark(benchmark_searcher.search, "Python 向量数据库", top_k=5)


def test_bench_search_top20(benchmark, benchmark_searcher):
    """top_k=20 检索耗时."""
    benchmark(benchmark_searcher.search, "Rust 编程语言 GPU", top_k=20)


def test_bench_search_cold_start(benchmark, benchmark_searcher):
    """冷启动检索耗时."""
    benchmark(benchmark_searcher.search, "GPU 加速 深度学习 嵌入", top_k=10)