def from_dtype(dtype):
if dtype.fields is None:
try:
basetype = FROM_DTYPE[dtype]
except KeyError:
m = re_typestr.match(dtype.str)
if not m:
raise NotImplementedError(dtype)
groups = m.groups()
typecode = groups[0]
if typecode == 'U':
# unicode
if dtype.byteorder not in '=|':
raise NotImplementedError("Does not support non-native "
"byteorder")
count = dtype.itemsize // sizeof_unicode_char
assert count == int(groups[1]), "Unicode char size mismatch"
return types.UnicodeCharSeq(count)
elif typecode == 'S':
# char
count = dtype.itemsize
assert count == int(groups[1]), "Char size mismatch"
return types.CharSeq(count)
raise NotImplementedError(dtype)
return basetype
else:
return from_struct_dtype(dtype)
def test_from_dtype(self):
rec = numpy_support.from_dtype(recordtype)
self.assertEqual(rec.typeof('a'), types.float64)
self.assertEqual(rec.typeof('b'), types.int16)
self.assertEqual(rec.typeof('c'), types.complex64)
self.assertEqual(rec.typeof('d'), types.UnicodeCharSeq(5))
self.assertEqual(rec.offset('a'), recordtype.fields['a'][1])
self.assertEqual(rec.offset('b'), recordtype.fields['b'][1])
self.assertEqual(rec.offset('c'), recordtype.fields['c'][1])
self.assertEqual(rec.offset('d'), recordtype.fields['d'][1])
self.assertEqual(recordtype.itemsize, rec.size)
def test_string_types(self):
"""
Test from_dtype() and as_dtype() with the character string types.
"""
def check(typestring, numba_type):
# Only native ordering and alignment is supported
dtype = np.dtype(typestring)
self.assertEqual(numpy_support.from_dtype(dtype), numba_type)
self.assertEqual(dtype, numpy_support.as_dtype(numba_type))
check('S10', types.CharSeq(10))
check('a11', types.CharSeq(11))
check('U12', types.UnicodeCharSeq(12))
