Add test case for '--extend-yaml-scientific-notation' cli option

Author: Maximilian Sander

tests/data/expected/main/jsonschema/scientific_notation.py

@@ -0,0 +1,13 @@
+# generated by datamodel-codegen:
+#   filename:  scientific_notation.json
+#   timestamp: 2025-03-28T00:00:00+00:00
+
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import BaseModel, Field
+
+
+class Model(BaseModel):
+    test: Optional[float] = Field(1e-05, description='Testcase', title='Test')

tests/data/jsonschema/scientific_notation.json

@@ -0,0 +1,12 @@
+{
+    "$schema": "http://json-schema.org/draft-07/schema#",
+    "type": "object",
+    "properties": {
+        "test": {
+            "type": "number",
+            "title": "Test",
+            "description": "Testcase",
+            "default": 1e-5
+        }
+    }
+}

tests/main/jsonschema/test_main_jsonschema.py

@@ -3114,3 +3114,22 @@ def test_main_jsonschema_field_has_same_name(output_model: str, expected_output:
         ])
         assert return_code == Exit.OK
         assert output_file.read_text() == (EXPECTED_JSON_SCHEMA_PATH / expected_output).read_text()
+
+@freeze_time("2025-03-28")
+def test_main_scientific_notation() -> None:
+    with TemporaryDirectory() as output_dir:
+        output_file: Path = Path(output_dir) / "scientific_notation.py"
+        expected_output: Path = EXPECTED_JSON_SCHEMA_PATH / "scientific_notation.py"
+        return_code: Exit = main([
+            "--extend-yaml-scientifc-notation",
+            "--input",
+            str(JSON_SCHEMA_DATA_PATH / "scientific_notation.json"),
+            "--output",
+            str(output_file),
+            "--input-file-type",
+            "jsonschema",
+            "--output-model-type",
+            "pydantic_v2.BaseModel",
+        ])
+        assert return_code == Exit.OK
+        assert output_file.read_text() == (EXPECTED_JSON_SCHEMA_PATH / expected_output).read_text()