datamodel-code-generator 0.28.5__tar.gz → 0.30.0__tar.gz

{datamodel_code_generator-0.28.5 → datamodel_code_generator-0.30.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datamodel-code-generator
-Version: 0.28.5
+Version: 0.30.0
 Summary: Datamodel Code Generator
 Project-URL: Homepage, https://github.com/koxudaxi/datamodel-code-generator
 Project-URL: Source, https://github.com/koxudaxi/datamodel-code-generator
@@ -21,7 +21,7 @@ Requires-Python: >=3.9
 Requires-Dist: argcomplete<4,>=2.10.1
 Requires-Dist: black>=19.10b0
 Requires-Dist: genson<2,>=1.2.1
-Requires-Dist: inflect<6,>=4.1
+Requires-Dist: inflect<8,>=4.1
 Requires-Dist: isort<7,>=4.3.21
 Requires-Dist: jinja2<4,>=2.10.1
 Requires-Dist: packaging
@@ -512,6 +512,8 @@ Model customization:
                         Choose Datetime class between AwareDatetime, NaiveDatetime or
                         datetime. Each output model has its default mapping (for example
                         pydantic: datetime, dataclass: str, ...)
+  --parent-scoped-naming
+                        Set name of models defined inline from the parent model
   --reuse-model         Reuse models on the field when a module has the model with the same
                         content
   --target-python-version {3.9,3.10,3.11,3.12,3.13}

{datamodel_code_generator-0.28.5 → datamodel_code_generator-0.30.0}/README.md

@@ -462,6 +462,8 @@ Model customization:
                         Choose Datetime class between AwareDatetime, NaiveDatetime or
                         datetime. Each output model has its default mapping (for example
                         pydantic: datetime, dataclass: str, ...)
+  --parent-scoped-naming
+                        Set name of models defined inline from the parent model
   --reuse-model         Reuse models on the field when a module has the model with the same
                         content
   --target-python-version {3.9,3.10,3.11,3.12,3.13}

{datamodel_code_generator-0.28.5 → datamodel_code_generator-0.30.0}/pyproject.toml

@@ -32,7 +32,7 @@ dependencies = [
   "argcomplete>=2.10.1,<4",
   "black>=19.10b0",
   "genson>=1.2.1,<2",
-  "inflect>=4.1,<6",
+  "inflect>=4.1,<8",
   "isort>=4.3.21,<7",
   "jinja2>=2.10.1,<4",
   "packaging",

{datamodel_code_generator-0.28.5 → datamodel_code_generator-0.30.0}/src/datamodel_code_generator/__init__.py

@@ -287,6 +287,7 @@ def generate(  # noqa: PLR0912, PLR0913, PLR0914, PLR0915
     keyword_only: bool = False,
     no_alias: bool = False,
     formatters: list[Formatter] = DEFAULT_FORMATTERS,
+    parent_scoped_naming: bool = False,
 ) -> None:
     remote_text_cache: DefaultPutDict[str, str] = DefaultPutDict()
     if isinstance(input_, str):
@@ -482,6 +483,8 @@ def generate(  # noqa: PLR0912, PLR0913, PLR0914, PLR0915
         keyword_only=keyword_only,
         no_alias=no_alias,
         formatters=formatters,
+        encoding=encoding,
+        parent_scoped_naming=parent_scoped_naming,
         **kwargs,
     )
 

{datamodel_code_generator-0.28.5 → datamodel_code_generator-0.30.0}/src/datamodel_code_generator/__main__.py

@@ -314,6 +314,7 @@ class Config(BaseModel):
     keyword_only: bool = False
     no_alias: bool = False
     formatters: list[Formatter] = DEFAULT_FORMATTERS
+    parent_scoped_naming: bool = False
 
     def merge_args(self, args: Namespace) -> None:
         set_args = {f: getattr(args, f) for f in self.get_fields() if getattr(args, f) is not None}
@@ -329,7 +330,7 @@ class Config(BaseModel):
             setattr(self, field_name, getattr(parsed_args, field_name))
 
 
-def _get_pyproject_toml_config(source: Path) -> dict[str, Any] | None:
+def _get_pyproject_toml_config(source: Path) -> dict[str, Any]:
     """Find and return the [tool.datamodel-codgen] section of the closest
     pyproject.toml if it exists.
     """
@@ -339,14 +340,20 @@ def _get_pyproject_toml_config(source: Path) -> dict[str, Any] | None:
         if (current_path / "pyproject.toml").is_file():
             pyproject_toml = load_toml(current_path / "pyproject.toml")
             if "datamodel-codegen" in pyproject_toml.get("tool", {}):
-                return pyproject_toml["tool"]["datamodel-codegen"]
+                pyproject_config = pyproject_toml["tool"]["datamodel-codegen"]
+                # Convert options from kebap- to snake-case
+                pyproject_config = {k.replace("-", "_"): v for k, v in pyproject_config.items()}
+                # Replace US-american spelling if present (ignore if british spelling is present)
+                if "capitalize_enum_members" in pyproject_config and "capitalise_enum_members" not in pyproject_config:
+                    pyproject_config["capitalise_enum_members"] = pyproject_config.pop("capitalize_enum_members")
+                return pyproject_config
 
         if (current_path / ".git").exists():
             # Stop early if we see a git repository root.
-            return None
+            return {}
 
         current_path = current_path.parent
-    return None
+    return {}
 
 
 def main(args: Sequence[str] | None = None) -> Exit:  # noqa: PLR0911, PLR0912, PLR0915
@@ -367,13 +374,9 @@ def main(args: Sequence[str] | None = None) -> Exit:  # noqa: PLR0911, PLR0912,
         sys.exit(0)
 
     pyproject_config = _get_pyproject_toml_config(Path.cwd())
-    if pyproject_config is not None:
-        pyproject_toml = {k.replace("-", "_"): v for k, v in pyproject_config.items()}
-    else:
-        pyproject_toml = {}
 
     try:
-        config = Config.parse_obj(pyproject_toml)
+        config = Config.parse_obj(pyproject_config)
         config.merge_args(namespace)
     except Error as e:
         print(e.message, file=sys.stderr)  # noqa: T201
@@ -523,6 +526,7 @@ def main(args: Sequence[str] | None = None) -> Exit:  # noqa: PLR0911, PLR0912,
             keyword_only=config.keyword_only,
             no_alias=config.no_alias,
             formatters=config.formatters,
+            parent_scoped_naming=config.parent_scoped_naming,
         )
     except InvalidClassNameError as e:
         print(f"{e} You have to set `--class-name` option", file=sys.stderr)  # noqa: T201

{datamodel_code_generator-0.28.5 → datamodel_code_generator-0.30.0}/src/datamodel_code_generator/arguments.py

@@ -202,6 +202,12 @@ model_options.add_argument(
     choices=[i.value for i in DatetimeClassType],
     default=None,
 )
+model_options.add_argument(
+    "--parent-scoped-naming",
+    help="Set name of models defined inline from the parent model",
+    action="store_true",
+    default=None,
+)
 
 # ======================================================================================
 # Typing options for generated models

{datamodel_code_generator-0.28.5 → datamodel_code_generator-0.30.0}/src/datamodel_code_generator/model/template/pydantic_v2/BaseModel.jinja2

@@ -1,3 +1,11 @@
+{% if base_class != "BaseModel" and "," not in base_class and not fields and not config -%}
+
+{# if this is just going to be `class Foo(Bar): pass`, then might as well just make Foo
+an alias for Bar: every pydantic model class consumes considerable memory. #}
+{{ class_name }} = {{ base_class }}
+
+{% else -%}
+
 {% for decorator in decorators -%}
 {{ decorator }}
 {% endfor -%}
@@ -37,3 +45,5 @@ class {{ class_name }}({{ base_class }}):{% if comment is defined %}  # {{ comme
     {{ method }}
 {%- endfor -%}
 {%- endfor -%}
+
+{%- endif %}

{datamodel_code_generator-0.28.5 → datamodel_code_generator-0.30.0}/src/datamodel_code_generator/parser/base.py

@@ -377,6 +377,7 @@ class Parser(ABC):
         keyword_only: bool = False,
         no_alias: bool = False,
         formatters: list[Formatter] = DEFAULT_FORMATTERS,
+        parent_scoped_naming: bool = False,
     ) -> None:
         self.keyword_only = keyword_only
         self.data_type_manager: DataTypeManager = data_type_manager_type(
@@ -463,6 +464,7 @@ class Parser(ABC):
             remove_special_field_name_prefix=remove_special_field_name_prefix,
             capitalise_enum_members=capitalise_enum_members,
             no_alias=no_alias,
+            parent_scoped_naming=parent_scoped_naming,
         )
         self.class_name: str | None = class_name
         self.wrap_string_literal: bool | None = wrap_string_literal
@@ -949,10 +951,14 @@ class Parser(ABC):
                             model_field.constraints = ConstraintsBase.merge_constraints(
                                 root_type_field.constraints, model_field.constraints
                             )
-                        if isinstance(
-                            root_type_field,
-                            pydantic_model.DataModelField,
-                        ) and not model_field.extras.get("discriminator"):
+                        if (
+                            isinstance(
+                                root_type_field,
+                                pydantic_model.DataModelField,
+                            )
+                            and not model_field.extras.get("discriminator")
+                            and not any(t.is_list for t in model_field.data_type.data_types)
+                        ):
                             discriminator = root_type_field.extras.get("discriminator")
                             if discriminator:
                                 model_field.extras["discriminator"] = discriminator
@@ -1148,7 +1154,7 @@ class Parser(ABC):
                 else:
                     r.append(item)
 
-            r = r[:-2] + [f"{r[-2]}.{r[-1]}"]
+            r = [*r[:-2], f"{r[-2]}.{r[-1]}"]
             return tuple(r)
 
         results = {process(k): v for k, v in results.items()}
@@ -1184,6 +1190,24 @@ class Parser(ABC):
                 class_name=True,
             ).name
 
+    def __alias_shadowed_imports(  # noqa: PLR6301
+        self,
+        models: list[DataModel],
+        all_model_field_names: set[str],
+    ) -> None:
+        for model in models:
+            for model_field in model.fields:
+                if model_field.data_type.type in all_model_field_names:
+                    alias = model_field.data_type.type + "_aliased"
+                    model_field.data_type.type = alias
+                    if model_field.data_type.import_:  # pragma: no cover
+                        model_field.data_type.import_ = Import(
+                            from_=model_field.data_type.import_.from_,
+                            import_=model_field.data_type.import_.import_,
+                            alias=alias,
+                            reference_path=model_field.data_type.import_.reference_path,
+                        )
+
     def parse(  # noqa: PLR0912, PLR0914, PLR0915
         self,
         with_import: bool | None = True,  # noqa: FBT001, FBT002
@@ -1278,6 +1302,7 @@ class Parser(ABC):
             all_module_fields = {field.name for model in models for field in model.fields if field.name is not None}
             scoped_model_resolver = ModelResolver(exclude_names=all_module_fields)
 
+            self.__alias_shadowed_imports(models, all_module_fields)
             self.__override_required_field(models)
             self.__replace_unique_list_to_set(models)
             self.__change_from_import(models, imports, scoped_model_resolver, init)

{datamodel_code_generator-0.28.5 → datamodel_code_generator-0.30.0}/src/datamodel_code_generator/parser/graphql.py

@@ -154,6 +154,7 @@ class GraphQLParser(Parser):
         keyword_only: bool = False,
         no_alias: bool = False,
         formatters: list[Formatter] = DEFAULT_FORMATTERS,
+        parent_scoped_naming: bool = False,
     ) -> None:
         super().__init__(
             source=source,
@@ -228,6 +229,7 @@ class GraphQLParser(Parser):
             keyword_only=keyword_only,
             no_alias=no_alias,
             formatters=formatters,
+            parent_scoped_naming=parent_scoped_naming,
         )
 
         self.data_model_scalar_type = data_model_scalar_type

{datamodel_code_generator-0.28.5 → datamodel_code_generator-0.30.0}/src/datamodel_code_generator/parser/jsonschema.py

@@ -423,6 +423,7 @@ class JsonSchemaParser(Parser):
         keyword_only: bool = False,
         no_alias: bool = False,
         formatters: list[Formatter] = DEFAULT_FORMATTERS,
+        parent_scoped_naming: bool = False,
     ) -> None:
         super().__init__(
             source=source,
@@ -497,6 +498,7 @@ class JsonSchemaParser(Parser):
             keyword_only=keyword_only,
             no_alias=no_alias,
             formatters=formatters,
+            parent_scoped_naming=parent_scoped_naming,
         )
 
         self.remote_object_cache: DefaultPutDict[str, dict[str, Any]] = DefaultPutDict()

{datamodel_code_generator-0.28.5 → datamodel_code_generator-0.30.0}/src/datamodel_code_generator/parser/openapi.py

@@ -216,6 +216,7 @@ class OpenAPIParser(JsonSchemaParser):
         keyword_only: bool = False,
         no_alias: bool = False,
         formatters: list[Formatter] = DEFAULT_FORMATTERS,
+        parent_scoped_naming: bool = False,
     ) -> None:
         super().__init__(
             source=source,
@@ -290,6 +291,7 @@ class OpenAPIParser(JsonSchemaParser):
             keyword_only=keyword_only,
             no_alias=no_alias,
             formatters=formatters,
+            parent_scoped_naming=parent_scoped_naming,
         )
         self.open_api_scopes: list[OpenAPIScope] = openapi_scopes or [OpenAPIScope.Schemas]
 

{datamodel_code_generator-0.28.5 → datamodel_code_generator-0.30.0}/src/datamodel_code_generator/reference.py

@@ -318,6 +318,7 @@ class ModelResolver:  # noqa: PLR0904
         capitalise_enum_members: bool = False,  # noqa: FBT001, FBT002
         no_alias: bool = False,  # noqa: FBT001, FBT002
         remove_suffix_number: bool = False,  # noqa: FBT001, FBT002
+        parent_scoped_naming: bool = False,  # noqa: FBT001, FBT002
     ) -> None:
         self.references: dict[str, Reference] = {}
         self._current_root: Sequence[str] = []
@@ -351,6 +352,7 @@ class ModelResolver:  # noqa: PLR0904
         self._base_path: Path = base_path or Path.cwd()
         self._current_base_path: Path | None = self._base_path
         self.remove_suffix_number: bool = remove_suffix_number
+        self.parent_scoped_naming = parent_scoped_naming
 
     @property
     def current_base_path(self) -> Path | None:
@@ -512,6 +514,19 @@ class ModelResolver:  # noqa: PLR0904
         self.references[path] = reference
         return reference
 
+    def _check_parent_scope_option(self, name: str, path: Sequence[str]) -> str:
+        if self.parent_scoped_naming:
+            parent_reference = None
+            parent_path = path[:-1]
+            while parent_path:
+                parent_reference = self.references.get(self.join_path(parent_path))
+                if parent_reference is not None:
+                    break
+                parent_path = parent_path[:-1]
+            if parent_reference:
+                name = f"{parent_reference.name}_{name}"
+        return name
+
     def add(  # noqa: PLR0913
         self,
         path: Sequence[str],
@@ -533,6 +548,7 @@ class ModelResolver:  # noqa: PLR0904
         name = original_name
         duplicate_name: str | None = None
         if class_name:
+            name = self._check_parent_scope_option(name, path)
             name, duplicate_name = self.get_class_name(
                 name=name,
                 unique=unique,

datamodel_code_generator-0.30.0/tests/data/expected/main/json/json_reuse_model_pydantic2.py

@@ -0,0 +1,26 @@
+# generated by datamodel-codegen:
+#   filename:  duplicate_models.json
+#   timestamp: 2019-07-26T00:00:00+00:00
+
+from __future__ import annotations
+
+from pydantic import BaseModel, Field
+
+
+class ArmRight(BaseModel):
+    Joint_1: int = Field(..., alias='Joint 1')
+    Joint_2: int = Field(..., alias='Joint 2')
+    Joint_3: int = Field(..., alias='Joint 3')
+
+
+ArmLeft = ArmRight
+
+
+class Head(BaseModel):
+    Joint_1: int = Field(..., alias='Joint 1')
+
+
+class Model(BaseModel):
+    Arm_Right: ArmRight = Field(..., alias='Arm Right')
+    Arm_Left: ArmLeft = Field(..., alias='Arm Left')
+    Head_1: Head = Field(..., alias='Head')

{datamodel_code_generator-0.28.5 → datamodel_code_generator-0.30.0}/tests/data/expected/main/openapi/discriminator/in_array_collapse_root_models.py

@@ -30,6 +30,4 @@ class UpdateObjectRequest(ObjectBase):
 
 
 class Demo(BaseModel):
-    myArray: List[Union[ObjectBase, CreateObjectRequest, UpdateObjectRequest]] = Field(
-        ..., discriminator='type'
-    )
+    myArray: List[Union[ObjectBase, CreateObjectRequest, UpdateObjectRequest]]

datamodel_code_generator-0.30.0/tests/data/expected/main/openapi/duplicate_models2.py

@@ -0,0 +1,42 @@
+# generated by datamodel-codegen:
+#   filename:  duplicate_models2.yaml
+#   timestamp: 2019-07-26T00:00:00+00:00
+
+from __future__ import annotations
+
+from enum import Enum
+from typing import Optional
+
+from pydantic import BaseModel
+
+
+class PetType(Enum):
+    pet = 'pet'
+
+
+class PetDetails(BaseModel):
+    race: Optional[str] = None
+
+
+class Pet(BaseModel):
+    id: int
+    name: str
+    tag: Optional[str] = None
+    type: PetType
+    details: Optional[PetDetails] = None
+
+
+class CarType(Enum):
+    car = 'car'
+
+
+class CarDetails(BaseModel):
+    brand: Optional[str] = None
+
+
+class Car(BaseModel):
+    id: int
+    name: str
+    tag: Optional[str] = None
+    type: CarType
+    details: Optional[CarDetails] = None

datamodel_code_generator-0.30.0/tests/data/expected/main/openapi/shadowed_imports.py

@@ -0,0 +1,15 @@
+# generated by datamodel-codegen:
+#   filename:  shadowed_imports.yaml
+#   timestamp: 2019-07-26T00:00:00+00:00
+
+from __future__ import annotations
+
+from datetime import date as date_aliased
+from typing import Optional
+
+from pydantic import BaseModel, Field
+
+
+class MarketingOptIn(BaseModel):
+    optedIn: Optional[bool] = Field(None, examples=[False])
+    date: Optional[date_aliased] = Field(None, examples=['2018-04-26T17:03:25.155Z'])

datamodel_code_generator-0.30.0/tests/data/openapi/duplicate_models2.yaml

@@ -0,0 +1,72 @@
+openapi: "3.0.0"
+info:
+  version: 1.0.0
+  title: Swagger Petstore
+  license:
+    name: MIT
+servers:
+  - url: http://petstore.swagger.io/v1
+paths:
+  /pets:
+    get:
+      summary: Get pet
+      operationId: getPets
+      responses:
+        '200':
+          content:
+            application/json:
+              schema:
+                $ref: "#/components/schemas/Pet"
+  /cars:
+    get:
+      summary: Get car
+      operationId: getCar
+      responses:
+        '200':
+          content:
+            application/json:
+              schema:
+                $ref: "#/components/schemas/Cars"
+
+components:
+  schemas:
+    Pet:
+      required:
+        - id
+        - name
+        - type
+      properties:
+        id:
+          type: integer
+          format: int64
+        name:
+          type: string
+        tag:
+          type: string
+        type:
+          type: string
+          enum: [ 'pet' ]
+        details:
+          type: object
+          properties:
+            race: { type: string }
+    Car:
+      required:
+        - id
+        - name
+        - type
+      properties:
+        id:
+          type: integer
+          format: int64
+        name:
+          type: string
+        tag:
+          type: string
+        type:
+          type: string
+          enum: [ 'car' ]
+        details:
+          type: object
+          properties:
+            brand: { type: string }

datamodel_code_generator-0.30.0/tests/data/openapi/shadowed_imports.yaml

@@ -0,0 +1,32 @@
+openapi: 3.0.0
+info:
+  title: REST API
+  version: 0.0.1
+servers:
+  - url: https://api.something.com/1
+components:
+  schemas:
+    marketingOptIn:
+      type: object
+      properties:
+        optedIn:
+          type: boolean
+          example: false
+        date:
+          type: string
+          format: date
+          example: '2018-04-26T17:03:25.155Z'
+paths:
+  /actions/:
+    get:
+      parameters:
+        - name: due
+          in: query
+          description: A due date for the card
+          required: false
+          schema:
+            type: string
+            format: date
+      responses:
+        '200':
+          description: Success

{datamodel_code_generator-0.28.5 → datamodel_code_generator-0.30.0}/tests/main/openapi/test_main_openapi.py

@@ -2568,3 +2568,43 @@ def test_main_openapi_referenced_default() -> None:
         ])
         assert return_code == Exit.OK
         assert output_file.read_text() == (EXPECTED_OPENAPI_PATH / "referenced_default.py").read_text()
+
+
+@freeze_time("2019-07-26")
+def test_duplicate_models() -> None:
+    with TemporaryDirectory() as output_dir:
+        output_file: Path = Path(output_dir) / "output.py"
+        return_code: Exit = main([
+            "--input",
+            str(OPEN_API_DATA_PATH / "duplicate_models2.yaml"),
+            "--output",
+            str(output_file),
+            "--use-operation-id-as-name",
+            "--openapi-scopes",
+            "paths",
+            "schemas",
+            "parameters",
+            "--output-model-type",
+            "pydantic_v2.BaseModel",
+            "--parent-scoped-naming",
+        ])
+        assert return_code == Exit.OK
+        assert output_file.read_text() == (EXPECTED_OPENAPI_PATH / "duplicate_models2.py").read_text()
+
+
+@freeze_time("2019-07-26")
+def test_main_openapi_shadowed_imports() -> None:
+    with TemporaryDirectory() as output_dir:
+        output_file: Path = Path(output_dir) / "output.py"
+        return_code: Exit = main([
+            "--input",
+            str(OPEN_API_DATA_PATH / "shadowed_imports.yaml"),
+            "--output",
+            str(output_file),
+            "--input-file-type",
+            "openapi",
+            "--output-model-type",
+            "pydantic_v2.BaseModel",
+        ])
+        assert return_code == Exit.OK
+        assert output_file.read_text() == (EXPECTED_OPENAPI_PATH / "shadowed_imports.py").read_text()

{datamodel_code_generator-0.28.5 → datamodel_code_generator-0.30.0}/tests/main/test_main_json.py

@@ -114,6 +114,25 @@ def test_main_json_reuse_model() -> None:
         assert output_file.read_text() == (EXPECTED_JSON_PATH / "json_reuse_model.py").read_text()
 
 
+@freeze_time("2019-07-26")
+def test_main_json_reuse_model_pydantic2() -> None:
+    with TemporaryDirectory() as output_dir:
+        output_file: Path = Path(output_dir) / "output.py"
+        return_code: Exit = main([
+            "--input",
+            str(JSON_DATA_PATH / "duplicate_models.json"),
+            "--output-model-type",
+            "pydantic_v2.BaseModel",
+            "--output",
+            str(output_file),
+            "--input-file-type",
+            "json",
+            "--reuse-model",
+        ])
+        assert return_code == Exit.OK
+        assert output_file.read_text() == (EXPECTED_JSON_PATH / "json_reuse_model_pydantic2.py").read_text()
+
+
 @freeze_time("2019-07-26")
 def test_simple_json_snake_case_field() -> None:
     with TemporaryDirectory() as output_dir:

{datamodel_code_generator-0.28.5 → datamodel_code_generator-0.30.0}/tests/test_main_kr.py

@@ -170,6 +170,70 @@ def test_pyproject() -> None:
         assert output_file.read_text() == (EXPECTED_MAIN_KR_PATH / "pyproject" / "output.py").read_text()
 
 
+@pytest.mark.parametrize("language", ["UK", "US"])
+def test_pyproject_respects_both_spellings_of_capitalize_enum_members_flag(language: str) -> None:
+    with TemporaryDirectory() as output_dir:
+        output_path = Path(output_dir)
+        pyproject_toml_data = f"""
+[tool.datamodel-codegen]
+capitali{"s" if language == "UK" else "z"}e-enum-members = true
+enable-version-header = false
+input-file-type = "jsonschema"
+"""
+        with (output_path / "pyproject.toml").open("w") as f:
+            f.write(pyproject_toml_data)
+
+        input_data = """
+{
+  "$schema": "http://json-schema.org/draft-07/schema#",
+  "definitions": {
+    "MyEnum": {
+      "enum": [
+        "MEMBER_1",
+        "member_2"
+      ]
+    }
+  }
+}
+"""
+        input_file = output_path / "schema.json"
+        with input_file.open("w") as f:
+            f.write(input_data)
+
+        expected_output = """# generated by datamodel-codegen:
+#   filename:  schema.json
+
+from __future__ import annotations
+
+from enum import Enum
+from typing import Any
+
+from pydantic import BaseModel
+
+
+class Model(BaseModel):
+    __root__: Any
+
+
+class MyEnum(Enum):
+    MEMBER_1 = 'MEMBER_1'
+    member_2 = 'member_2'
+"""
+
+        output_file: Path = output_path / "output.py"
+        return_code: Exit = main([
+            "--disable-timestamp",
+            "--input",
+            input_file.as_posix(),
+            "--output",
+            output_file.as_posix(),
+        ])
+        assert return_code == Exit.OK
+        assert output_file.read_text() == expected_output, (
+            f"\nExpected  output:\n{expected_output}\n\nGenerated output:\n{output_file.read_text()}"
+        )
+
+
 @pytest.mark.skipif(
     black.__version__.split(".")[0] == "19",
     reason="Installed black doesn't support the old style",