Function test_decimals

tests/functional/test_models.py:261–321

()

def test_decimals():
    class IsDecimal(Record, coerce=True, serializer="json"):
        number: Decimal

    class ListOfDecimal(Record, coerce=True, serializer="json"):
        numbers: List[Decimal]

    class OptionalListOfDecimal(Record, coerce=True, serializer="json"):
        numbers: List[Decimal] = None

    class OptionalListOfDecimal2(Record, coerce=True, serializer="json"):
        numbers: Optional[List[Decimal]]

    class TupleOfDecimal(Record, coerce=True, serializer="json"):
        numbers: Tuple[Decimal, ...]

    class SetOfDecimal(Record, coerce=True, serializer="json"):
        numbers: Set[Decimal]

    class MapOfDecimal(Record, coerce=True, serializer="json"):
        numbers: Mapping[str, Decimal]

    n1 = Decimal("1.31341324")
    assert IsDecimal.loads(IsDecimal(number=n1).dumps()).number == n1
    n2 = Decimal("3.41569")
    assert ListOfDecimal.loads(ListOfDecimal(numbers=[n1, n2]).dumps()).numbers == [
        n1,
        n2,
    ]
    assert (
        OptionalListOfDecimal.loads(OptionalListOfDecimal(numbers=None).dumps()).numbers
        is None
    )
    assert OptionalListOfDecimal.loads(
        OptionalListOfDecimal(numbers=[n2, n1]).dumps()
    ).numbers == [n2, n1]
    assert (
        OptionalListOfDecimal2.loads(
            OptionalListOfDecimal2(numbers=None).dumps()
        ).numbers
        is None
    )
    assert OptionalListOfDecimal2.loads(
        OptionalListOfDecimal2(numbers=[n1, n2]).dumps()
    ).numbers == [n1, n2]
    assert TupleOfDecimal.loads(TupleOfDecimal(numbers=(n1, n2)).dumps()).numbers == (
        n1,
        n2,
    )
    assert TupleOfDecimal.loads(TupleOfDecimal(numbers=(n2,)).dumps()).numbers == (n2,)
    assert SetOfDecimal.loads(SetOfDecimal(numbers={n1, n2}).dumps()).numbers == {
        n1,
        n2,
    }
    assert MapOfDecimal.loads(
        MapOfDecimal(numbers={"A": n1, "B": n2}).dumps()
    ).numbers == {"A": n1, "B": n2}