Source code for ase2sprkkr.bindings.xband.tests.alternative_types

""" 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)