From 519afdde5f863a907a7099c8dc9a492e3aabf1fc Mon Sep 17 00:00:00 2001 From: Matthew Farrellee Date: Tue, 22 Apr 2025 09:54:57 -0400 Subject: [PATCH] vllm unit test, check for exception on error --- tests/unit/providers/inference/test_remote_vllm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/unit/providers/inference/test_remote_vllm.py b/tests/unit/providers/inference/test_remote_vllm.py index 88399198d..491b2ba2e 100644 --- a/tests/unit/providers/inference/test_remote_vllm.py +++ b/tests/unit/providers/inference/test_remote_vllm.py @@ -107,8 +107,8 @@ async def test_register_model_checks_vllm(mock_openai_models_list, vllm_inferenc foo_model = Model(identifier="foo", provider_resource_id="foo", provider_id="vllm-inference") - await vllm_inference_adapter.register_model(foo_model) - mock_openai_models_list.assert_called() + with pytest.raises(ValueError): + await vllm_inference_adapter.register_model(foo_model) @pytest.mark.asyncio