示例#1
0
文件: test_conv.py 项目: minrk/h5py
    def test_obj2vlen_simple(self):
        """ Object to vlen (contiguous) """

        objarr = np.array(strings, dtype=vlen_dtype)

        destbuffer = np.ndarray(objarr.shape, dtype=np.uintp, buffer=objarr).copy()

        h5t.convert(obj_htype, vlen_htype, len(strings), destbuffer)

        for idx, val in enumerate(destbuffer):
            self.assertEqual(ctypes.string_at(int(val)), strings[idx])
示例#2
0
文件: test_conv.py 项目: minrk/h5py
    def test_obj2vlen_complex(self):
        """ Object to vlen (compound) """

        obj_ptr_size = h5t.PYTHON_OBJECT.get_size()
        vlen_ptr_size = vlen_htype.get_size()

        input_htype = h5t.create(h5t.COMPOUND, obj_ptr_size+4)
        input_htype.insert('a', 0, obj_htype)
        input_htype.insert('b', obj_ptr_size, h5t.STD_I32LE)

        output_htype = h5t.create(h5t.COMPOUND, vlen_ptr_size+4)
        output_htype.insert('a', 0, vlen_htype)
        output_htype.insert('b', vlen_ptr_size, h5t.STD_I32LE)

        objarr = np.ndarray((len(strings),), dtype=[('a', vlen_dtype), ('b', '<i4')])
        objarr['a'] = strings

        destbuffer = np.ndarray(objarr.shape, dtype=[('a', np.uintp), ('b', '<i4')], buffer=objarr).copy()

        h5t.convert(input_htype, output_htype, len(strings), destbuffer, destbuffer.copy())

        for idx, val in enumerate(destbuffer):
            self.assertEqual(ctypes.string_at(int(val[0])), strings[idx])