"""This module contains helper functions and data to deal with the problem, that
numpy arrays contains their own datatypes and thus e.g. an ``int`` may not be what
it seems"""
import numpy as np
import datetime as datetime
numpy_types = {int: np.int32, float: np.float64, complex: np.complex128}
""" Map to numpy types """
types_alternatives = {
int: (np.int64, np.int32),
np.int32: (int, np.int64),
float: (np.float64,),
np.float64: (float,),
complex: (np.complex128,),
np.complex128: (complex,),
bool: (np.bool_,),
datetime.datetime: (np.datetime64,),
}
"Map of types and sets of alternative types"
normalize_type_map = {}
""" Mapping of alternative types to the 'canonical ones'. """
# fill the map
for i in types_alternatives:
if i not in normalize_type_map:
normalize_type_map[i] = i
for j in types_alternatives[i]:
if j not in normalize_type_map:
normalize_type_map[j] = i
del i, j
allowed_types = {i: (*j, i) for i, j in types_alternatives.items()}
""" All types, that are allowed for a given type. I.e., the content of types_alternatives and the primary
type itself """
[docs]
def normalize_type(type):
"""Return the 'canonical type' for a given type.
I.e. it maps numpy internal types to standard python ones.
doctest:
>>> normalize_type(np.int64)
<class 'int'>
"""
return normalize_type_map.get(type, type)