tests/unit/test_gpu_gate.py

from agentic_search.retrieval.gpu_gate import inference_lock


def test_cuda_retrieval_models_share_one_process_lock() -> None:
    assert inference_lock("cuda") is inference_lock("cuda")
    assert inference_lock("cuda") is not inference_lock("cpu")