1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
"""Tests for YAML input file code generation."""
from __future__ import annotations
from typing import TYPE_CHECKING
import pytest
from datamodel_code_generator.__main__ import Exit
from tests.conftest import create_assert_file_content
from tests.main.conftest import (
EXPECTED_MAIN_PATH,
YAML_DATA_PATH,
run_main_and_assert,
)
if TYPE_CHECKING:
from pathlib import Path
assert_file_content = create_assert_file_content(EXPECTED_MAIN_PATH)
@pytest.mark.benchmark
@pytest.mark.cli_doc(
options=["--input-file-type"],
option_description="""Generate models from raw YAML sample data.
The `--input-file-type yaml` option treats the input file as **raw YAML data**
and automatically infers a JSON Schema from it.
**Note:** This is NOT for JSON Schema files written in YAML format.
For schema definition files (JSON Schema or OpenAPI), use `--input-file-type jsonschema`
or `--input-file-type openapi` instead, regardless of whether the file is in JSON or YAML format.""",
input_schema="yaml/pet.yaml",
cli_args=["--input-file-type", "yaml"],
golden_output="yaml.py",
)
def test_main_yaml(output_file: Path) -> None:
"""Generate models from raw YAML sample data.
The `--input-file-type yaml` option treats the input file as **raw YAML data**
and automatically infers a JSON Schema from it.
**Note:** This is NOT for JSON Schema files written in YAML format.
For schema definition files (JSON Schema or OpenAPI), use `--input-file-type jsonschema`
or `--input-file-type openapi` instead, regardless of whether the file is in JSON or YAML format.
"""
run_main_and_assert(
input_path=YAML_DATA_PATH / "pet.yaml",
output_path=output_file,
input_file_type="yaml",
assert_func=assert_file_content,
)
def test_main_yaml_invalid_root_list(output_file: Path, capsys: pytest.CaptureFixture[str]) -> None:
"""Test YAML file with list as root element fails with invalid file format error."""
run_main_and_assert(
input_path=YAML_DATA_PATH / "invalid_root_list.yaml",
output_path=output_file,
input_file_type="yaml",
expected_exit=Exit.ERROR,
capsys=capsys,
expected_stderr_contains="Invalid file format",
)
def test_main_yaml_deprecated_bool(output_file: Path) -> None:
"""Test YAML file with deprecated bool syntax emits deprecation warning."""
with pytest.warns(DeprecationWarning, match=r"YAML bool 'True' is deprecated"):
run_main_and_assert(
input_path=YAML_DATA_PATH / "deprecated_bool.yaml",
output_path=output_file,
input_file_type="openapi",
)
def test_main_yaml_scientific_notation(output_file: Path) -> None:
"""Test YAML file with scientific notation default values (issue #1955).
Scientific notation without decimal point (e.g., 1e-5) should be parsed as float,
not as string.
"""
run_main_and_assert(
input_path=YAML_DATA_PATH / "scientific_notation.yaml",
output_path=output_file,
input_file_type="jsonschema",
extra_args=["--output-model-type", "pydantic_v2.BaseModel"],
assert_func=assert_file_content,
expected_file="yaml/scientific_notation.py",
)
|
