diff --git a/tests/v1/sample/test_sampling_params_e2e.py b/tests/v1/sample/test_sampling_params_e2e.py index e47f13f05316..2ea01e667c6b 100644 --- a/tests/v1/sample/test_sampling_params_e2e.py +++ b/tests/v1/sample/test_sampling_params_e2e.py @@ -121,6 +121,10 @@ def test_allowed_token_ids(model): PROMPT, SamplingParams(allowed_token_ids=allowed_token_ids)) assert output[0].outputs[0].token_ids[-1] == TOKEN_ID + # Reject empty allowed_token_ids. + with pytest.raises(ValueError): + _ = model.generate(PROMPT, SamplingParams(allowed_token_ids=[])) + # Reject negative token id. with pytest.raises(ValueError): _ = model.generate(PROMPT, SamplingParams(allowed_token_ids=[-1]))