graphinate.tools.converters

Complexity

Total Complexity

2

Size/Duplication

Total Lines	28
Duplicated Lines	0 %

Importance

Changes

0

2 Functions

Rating	Name	Duplication	Size	Complexity
A	infnum_to_str()	0	2	1
A	str_to_infnum()	0	2	1

import decimal
import math
from types import MappingProxyType
from typing import NewType, Union

InfNumber = NewType("InfNumber", Union[float, int, decimal.Decimal])

INFINITY_MAPPER = MappingProxyType({
    'Infinity': math.inf,
    '-Infinity': -math.inf
})

MATH_INF_MAPPER = MappingProxyType({
    math.inf: 'Infinity',
    -math.inf: '-Infinity'
})


def str_to_infnum(value: str):
    return INFINITY_MAPPER.get(value, value)


def infnum_to_str(value: InfNumber):
    return MATH_INF_MAPPER.get(value, value)


__all__ = ['InfNumber', 'infnum_to_str', 'str_to_infnum']