test: move test

Author: watermarkhu

tests/test_extension.py

@@ -55,6 +55,10 @@ def regular_method(self):
 
     class RegularClass(object):
         regular_attr = 1
+
+    class AliasClass(BaseModel):
+        internal_name: str = Field(default="test", serialization_alias="external_name")
+        regular_field: int = Field(default=42)
 """
 
 
@@ -65,7 +69,7 @@ def test_extension(analysis: str) -> None:
     with loader(
         "package",
         modules={"__init__.py": code},
-        extensions=Extensions(PydanticExtension(schema=True)),
+        extensions=Extensions(PydanticExtension(schema=True, serialize_by_alias=True)),
         search_sys_path=analysis == "dynamic",
     ) as package:
         assert package
@@ -82,6 +86,14 @@ def test_extension(analysis: str) -> None:
         schema = package.classes["ExampleModel"].extra["griffe_pydantic"]["schema"]
         assert schema.startswith('{\n  "description"')
 
+        assert "AliasClass" in package.classes
+        assert package.classes["AliasClass"].labels == {"pydantic-model"}
+
+        fields = package.classes["AliasClass"].extra["griffe_pydantic"]["fields"]()
+        assert "internal_name" not in fields
+        assert "regular_field" in fields
+        assert "external_name" in fields
+
 
 def test_imported_models() -> None:
     """Test the extension with imported models."""
@@ -407,28 +419,3 @@ class Model(BaseModel):
         assert "internal_name" in fields
         assert "regular_field" in fields
         assert "external_name" not in fields
-
-
-@pytest.mark.parametrize("analysis", ["static", "dynamic"])
-def test_serialize_by_alias_enabled(analysis: str) -> None:
-    """Test that serialize_by_alias extension setting uses serialization_alias as the field name."""
-    code = """
-    from pydantic import BaseModel, Field
-
-    class Model(BaseModel):
-        internal_name: str = Field(default="test", serialization_alias="external_name")
-        regular_field: int = Field(default=42)
-    """
-    with temporary_visited_package(
-        "package",
-        modules={"__init__.py": code},
-        extensions=Extensions(PydanticExtension(schema=False, serialize_by_alias=True)),
-    ) as package:
-        
-        model = package["Model"]
-        assert model.labels == {"pydantic-model"}
-
-        fields = model.extra["griffe_pydantic"]["fields"]()
-        assert "internal_name" not in fields
-        assert "regular_field" in fields
-        assert "external_name" in fields