Exemple #1
0
 def test_assignment_arrfunc(self):
     af = _lowlevel.make_arrfunc_from_assignment(
                 ndt.float32, ndt.int64, "nocheck")
     self.assertEqual(nd.as_py(af.proto), ndt.type("(int64) -> float32"))
     a = nd.array(1234, type=ndt.int64)
     b = af(a)
     self.assertEqual(nd.type_of(b), ndt.float32)
     self.assertEqual(nd.as_py(b), 1234)
     # Instantiate as a strided kernel
     with _lowlevel.ckernel.CKernelBuilder() as ckb:
         meta = (ctypes.c_void_p * 2)()
         ectx = nd.eval_context()
         _lowlevel.arrfunc_instantiate(af, ckb, 0, ndt.float32, 0,
                                       [ndt.int64], [0], "strided",
                                       ectx._ectx_ptr)
         ck = ckb.ckernel(_lowlevel.ExprStridedOperation)
         # Do an assignment using ctypes
         i64 = (ctypes.c_int64 * 3)()
         for i, v in enumerate([3,7,21]):
             i64[i] = v
         pi64 = ctypes.pointer(i64)
         i64_stride = c_ssize_t(8)
         f32 = (ctypes.c_float * 3)()
         ck(ctypes.addressof(f32), 4,
                     ctypes.addressof(pi64), ctypes.pointer(i64_stride),
                     3)
         self.assertEqual([f32[i] for i in range(3)], [3,7,21])
Exemple #2
0
 def test_creation(self):
     af = nd.empty('arrfunc')
     self.assertEqual(nd.type_of(af).type_id, 'arrfunc')
     # Test there is a string version of a NULL arrfunc
     self.assertTrue(str(af) != '')
     self.assertEqual(nd.as_py(af.proto), ndt.type())
     # Test there is a string version of an initialized arrfunc
     af = _lowlevel.make_arrfunc_from_assignment(
                 ndt.float32, ndt.int64, "nocheck")
     self.assertTrue(str(af) != '')
     self.assertEqual(nd.as_py(af.proto), ndt.type("(int64) -> float32"))