test_prompt_validation.py

import pytest
import os

from vllm import LLM
from ...utils import models_path_prefix


def test_empty_prompt():
    llm = LLM(model=os.path.join(models_path_prefix, "gpt2"))
    with pytest.raises(ValueError, match='Prompt cannot be empty'):
        llm.generate([""])