Black

tests/test_pydantic_v1_models.py

@@ -1,12 +1,14 @@
 import pytest
 from fastapi import Body, FastAPI
+from fastapi._compat import PYDANTIC_V2
 from fastapi.exceptions import ResponseValidationError
 from fastapi.testclient import TestClient
-from pydantic import v1
 from typing_extensions import Annotated
 
 from tests.utils import needs_pydanticv2
 
+if PYDANTIC_V2:
+    from pydantic import v1
 
     class Item(v1.BaseModel):
         name: str
@@ -15,10 +17,22 @@ class Item(v1.BaseModel):
         tax: float | None = None
         tags: list = []
 
-
     class Model(v1.BaseModel):
         name: str
 
+else:
+    from pydantic import BaseModel
+
+    class Item(BaseModel):
+        name: str
+        description: str | None = None
+        price: float
+        tax: float | None = None
+        tags: list = []
+
+    class Model(BaseModel):
+        name: str
+
 
 app = FastAPI()