diff --git a/tests/test_ast_inference.py b/tests/test_ast_inference.py index f3482bb6c..70270f4d1 100644 --- a/tests/test_ast_inference.py +++ b/tests/test_ast_inference.py @@ -446,13 +446,14 @@ def test_infer_response_model_create_model_error(): func() from fastapi.utils import PYDANTIC_V2 - target = "pydantic.create_model" if PYDANTIC_V2 else "fastapi._compat.v1.create_model" - + target = ( + "pydantic.create_model" if PYDANTIC_V2 else "fastapi._compat.v1.create_model" + ) + with patch(target, side_effect=Exception("Boom")): assert infer_response_model_from_ast(func) is None - def test_contains_response() -> None: from fastapi.routing import _contains_response