1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
"""Tests for Pydantic constraint detection."""
from __future__ import annotations
import pytest
from datamodel_code_generator.model.pydantic.base_model import Constraints
from datamodel_code_generator.types import UnionIntFloat
@pytest.mark.parametrize(
("gt", "expected"),
[
(None, False),
(4, True),
(0, True),
(0.0, True),
],
)
def test_constraint(gt: float | None, expected: bool) -> None:
"""Test constraint detection with greater-than values."""
constraints = Constraints()
constraints.gt = UnionIntFloat(gt) if gt is not None else None
assert constraints.has_constraints == expected
|
