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