def test_buffer_iter(self): @jit def f(b): sum = 0 for x in b: sum += x return sum b = fromseq(range(10), int32) self.assertEqual(f(b), sum(range(10))) b = fromseq(range(11), int32) self.assertEqual(f(b), sum(range(11)))
def test_buffer_getitem(self): @jit def f(b): return b[1] b = fromseq([1, 2, 3], int32) self.assertEqual(f(b), 2)
def test_buffer_setitem(self): @jit def f(b): b[1] = 14 b = fromseq([1, 2, 3], int32) f(b) self.assertEqual(b[1], 14)
def test_buffer_setslice(self): @jit def f(b): b[1:-1:2] = 0 b = fromseq(range(10), int32) f(b) x = np.arange(10) f.py_func(x) self.assertTrue(np.all(list(b) == x))