Fix another test

tests/data/expected/openapi/disable_timestamp/simple/main.py

@@ -3,11 +3,11 @@
 
 from __future__ import annotations
 
-from typing import Optional, Union
+from typing import List, Optional, Union
 
 from fastapi import FastAPI, Path
 
-from .models import Error, Pets
+from .models import Error, Pet
 
 app = FastAPI(
     version='1.0.0',
@@ -19,9 +19,12 @@
 
 
 @app.get(
-    '/pets', response_model=Pets, responses={'default': {'model': Error}}, tags=['pets']
+    '/pets',
+    response_model=List[Pet],
+    responses={'default': {'model': Error}},
+    tags=['pets'],
 )
-def list_pets(limit: Optional[int] = None) -> Union[Pets, Error]:
+def list_pets(limit: Optional[int] = None) -> Union[List[Pet], Error]:
     """
     List all pets
     """
@@ -40,11 +43,11 @@ def create_pets() -> Optional[Error]:
 
 @app.get(
     '/pets/{petId}',
-    response_model=Pets,
+    response_model=List[Pet],
     responses={'default': {'model': Error}},
     tags=['pets'],
 )
-def show_pet_by_id(pet_id: str = Path(..., alias='petId')) -> Union[Pets, Error]:
+def show_pet_by_id(pet_id: str = Path(..., alias='petId')) -> Union[List[Pet], Error]:
     """
     Info for a specific pet
     """

tests/data/expected/openapi/disable_timestamp/simple/models.py

@@ -3,9 +3,9 @@
 
 from __future__ import annotations
 
-from typing import List, Optional
+from typing import Optional
 
-from pydantic import BaseModel, Field
+from pydantic import BaseModel
 
 
 class Pet(BaseModel):
@@ -14,10 +14,6 @@ class Pet(BaseModel):
     tag: Optional[str] = None
 
 
-class Pets(BaseModel):
-    __root__: List[Pet] = Field(..., description='list of pet')
-
-
 class Error(BaseModel):
     code: int
     message: str

tests/test_generate.py

@@ -127,6 +127,10 @@ def test_disable_timestamp(oas_file):
         expected_files = sorted(list(expected_dir.glob('*')))
         assert [f.name for f in output_files] == [f.name for f in expected_files]
         for output_file, expected_file in zip(output_files, expected_files):
+            print(expected_file)
+            print(expected_file.read_text())
+            print(output_file)
+            print(output_file.read_text())
             assert output_file.read_text() == expected_file.read_text()