def test_index(self):
pyfunc = tuple_index
cfunc = jit(nopython=True)(pyfunc)
p = Point(4, 5, 6)
for i in range(len(p)):
self.assertPreciseEqual(cfunc(p, i), pyfunc(p, i))
# test uintp indexing (because, e.g., parfor generates unsigned prange)
for i in range(len(p)):
self.assertPreciseEqual(cfunc(p, types.uintp(i)), pyfunc(p, i))
def test_index(self):
pyfunc = tuple_index
cfunc = jit(nopython=True)(pyfunc)
p = Point(4, 5, 6)
for i in range(len(p)):
self.assertPreciseEqual(cfunc(p, i), pyfunc(p, i))
# test uintp indexing (because, e.g., parfor generates unsigned prange)
for i in range(len(p)):
self.assertPreciseEqual(cfunc(p, types.uintp(i)), pyfunc(p, i))
def test_index(self):
pyfunc = tuple_index
cr = compile_isolated(pyfunc,
[types.UniTuple(types.int64, 3), types.int64])
tup = (4, 3, 6)
for i in range(len(tup)):
self.assertPreciseEqual(cr.entry_point(tup, i), tup[i])
# test negative indexing
for i in range(len(tup) + 1):
self.assertPreciseEqual(cr.entry_point(tup, -i), tup[-i])
# oob indexes, +ve then -ve
with self.assertRaises(IndexError) as raises:
cr.entry_point(tup, len(tup))
self.assertEqual("tuple index out of range", str(raises.exception))
with self.assertRaises(IndexError) as raises:
cr.entry_point(tup, -(len(tup) + 1))
self.assertEqual("tuple index out of range", str(raises.exception))
# Test empty tuple
cr = compile_isolated(pyfunc,
[types.UniTuple(types.int64, 0), types.int64])
with self.assertRaises(IndexError) as raises:
cr.entry_point((), 0)
self.assertEqual("tuple index out of range", str(raises.exception))
# test uintp indexing (because, e.g., parfor generates unsigned prange)
cr = compile_isolated(pyfunc,
[types.UniTuple(types.int64, 3), types.uintp])
for i in range(len(tup)):
self.assertPreciseEqual(cr.entry_point(tup, types.uintp(i)),
tup[i])
# With a compile-time static index (the code generation path is different)
pyfunc = tuple_index_static
for typ in (types.UniTuple(types.int64, 4),
types.Tuple(
(types.int64, types.int32, types.int64, types.int32))):
cr = compile_isolated(pyfunc, (typ, ))
tup = (4, 3, 42, 6)
self.assertPreciseEqual(cr.entry_point(tup), pyfunc(tup))
typ = types.UniTuple(types.int64, 1)
with self.assertTypingError():
cr = compile_isolated(pyfunc, (typ, ))
# test unpack, staticgetitem with imprecise type (issue #3895)
pyfunc = tuple_unpack_static_getitem_err
with self.assertTypingError() as raises:
cr = compile_isolated(pyfunc, ())
msg = ("Cannot infer the type of variable 'c', have imprecise type: "
"list(undefined).")
self.assertIn(msg, str(raises.exception))
def test_index(self):
pyfunc = tuple_index
cr = compile_isolated(pyfunc,
[types.UniTuple(types.int64, 3), types.int64])
tup = (4, 3, 6)
for i in range(len(tup)):
self.assertPreciseEqual(cr.entry_point(tup, i), tup[i])
# test negative indexing
for i in range(len(tup) + 1):
self.assertPreciseEqual(cr.entry_point(tup, -i), tup[-i])
# oob indexes, +ve then -ve
with self.assertRaises(IndexError) as raises:
cr.entry_point(tup, len(tup))
self.assertEqual("tuple index out of range", str(raises.exception))
with self.assertRaises(IndexError) as raises:
cr.entry_point(tup, -(len(tup) + 1))
self.assertEqual("tuple index out of range", str(raises.exception))
# Test empty tuple
cr = compile_isolated(pyfunc,
[types.UniTuple(types.int64, 0), types.int64])
with self.assertRaises(IndexError) as raises:
cr.entry_point((), 0)
self.assertEqual("tuple index out of range", str(raises.exception))
# test uintp indexing (because, e.g., parfor generates unsigned prange)
cr = compile_isolated(pyfunc,
[types.UniTuple(types.int64, 3), types.uintp])
for i in range(len(tup)):
self.assertPreciseEqual(cr.entry_point(tup, types.uintp(i)), tup[i])
# With a compile-time static index (the code generation path is different)
pyfunc = tuple_index_static
for typ in (types.UniTuple(types.int64, 4),
types.Tuple((types.int64, types.int32, types.int64, types.int32))):
cr = compile_isolated(pyfunc, (typ,))
tup = (4, 3, 42, 6)
self.assertPreciseEqual(cr.entry_point(tup), pyfunc(tup))
typ = types.UniTuple(types.int64, 1)
with self.assertTypingError():
cr = compile_isolated(pyfunc, (typ,))
# test unpack, staticgetitem with imprecise type (issue #3895)
pyfunc = tuple_unpack_static_getitem_err
with self.assertTypingError() as raises:
cr = compile_isolated(pyfunc, ())
msg = ("Cannot infer the type of variable 'c', have imprecise type: "
"list(undefined).")
self.assertIn(msg, str(raises.exception))
def gen_hash_compare_ops(key_type):
deref_voidptr = gen_deref_voidptr(key_type)
c_sig_hash = types.uintp(types.voidptr)
c_sig_eq = types.boolean(types.voidptr, types.voidptr)
@cfunc(c_sig_hash)
def hash_func_adaptor(voidptr_to_data):
obj = deref_voidptr(voidptr_to_data)
return hash(obj)
@cfunc(c_sig_eq)
def eq_func_adaptor(lhs_ptr, rhs_ptr):
lhs_str = deref_voidptr(lhs_ptr)
rhs_str = deref_voidptr(rhs_ptr)
return lhs_str == rhs_str
hasher_ptr = hash_func_adaptor.address
eq_ptr = eq_func_adaptor.address
return hasher_ptr, eq_ptr