コード例 #1
0
def zeros(dshape, ddesc=None):
    """Create an array and fill it with zeros.

    Parameters
    ----------
    dshape : datashape
        The datashape for the resulting array.

    ddesc : data descriptor instance
        This comes with the necessary info for storing the data.  If
        None, a DyND_DDesc will be used.

    Returns
    -------
    out : a concrete blaze array.

    """
    dshape = _normalize_dshape(dshape)

    if ddesc is None:
        ddesc = DyND_DDesc(nd.zeros(str(dshape), access='rw'))
        return Array(ddesc)
    if isinstance(ddesc, BLZ_DDesc):
        shape, dt = to_numpy(dshape)
        ddesc.blzarr = blz.zeros(
            shape, dt, rootdir=ddesc.path, mode=ddesc.mode, **ddesc.kwargs)
    elif isinstance(ddesc, HDF5_DDesc):
        obj = nd.as_numpy(nd.zeros(str(dshape)))
        with tb.open_file(ddesc.path, mode=ddesc.mode) as f:
            where, name = split_path(ddesc.datapath)
            f.create_earray(where, name, filters=ddesc.filters, obj=obj)
        ddesc.mode = 'a'  # change into 'a'ppend mode for further operations
    return Array(ddesc)
コード例 #2
0
 def test_access_zeros(self):
     a = nd.zeros(ndt.int32)
     self.assertEqual(a.access_flags, 'readwrite')
     a = nd.zeros(ndt.int32, access='rw')
     self.assertEqual(a.access_flags, 'readwrite')
     a = nd.zeros(ndt.int32, access='r')
     self.assertEqual(a.access_flags, 'immutable')
コード例 #3
0
 def test_access_zeros(self):
     a = nd.zeros(ndt.int32)
     self.assertEqual(a.access_flags, 'readwrite')
     a = nd.zeros(ndt.int32, access='rw')
     self.assertEqual(a.access_flags, 'readwrite')
     a = nd.zeros(ndt.int32, access='r')
     self.assertEqual(a.access_flags, 'immutable')
コード例 #4
0
 def test_squeeze_strided(self):
     # Simple strided array
     a = nd.zeros(1, 1, 3, 1, 2, 1, 1, 1, ndt.int32)
     self.assertEqual(a.shape, (1, 1, 3, 1, 2, 1, 1, 1))
     self.assertEqual(nd.squeeze(a).shape, (3, 1, 2))
     # Strip dimensions from the start
     a = nd.zeros(1, 3, ndt.float32)
     self.assertEqual(a.shape, (1, 3))
     self.assertEqual(nd.squeeze(a).shape, (3,))
     # Strip dimensions from the end
     a = nd.zeros(3, 1, ndt.float32)
     self.assertEqual(a.shape, (3, 1))
     self.assertEqual(nd.squeeze(a).shape, (3,))
コード例 #5
0
 def test_squeeze_strided(self):
     # Simple strided array
     a = nd.zeros(1, 1, 3, 1, 2, 1, 1, 1, ndt.int32)
     self.assertEqual(a.shape, (1, 1, 3, 1, 2, 1, 1, 1))
     self.assertEqual(nd.squeeze(a).shape, (3, 1, 2))
     # Strip dimensions from the start
     a = nd.zeros(1, 3, ndt.float32)
     self.assertEqual(a.shape, (1, 3))
     self.assertEqual(nd.squeeze(a).shape, (3, ))
     # Strip dimensions from the end
     a = nd.zeros(3, 1, ndt.float32)
     self.assertEqual(a.shape, (3, 1))
     self.assertEqual(nd.squeeze(a).shape, (3, ))
コード例 #6
0
ファイル: test_numpy_compat.py プロジェクト: Vasyka/hat
 def test_array_attributes(self):
     a = nd.zeros(3, 5, ndt.int32)
     # Property "ndim"
     self.assertEqual(a.ndim, 2)
     # Property "shape"
     self.assertEqual(a.shape, (3, 5))
     # Property "strides"
     self.assertEqual(a.strides, (20, 4))
     # Property "dtype"
     self.assertEqual(a.dtype, ndt.int32)
コード例 #7
0
 def test_array_attributes(self):
     a = nd.zeros(3, 5, ndt.int32)
     # Property "ndim"
     self.assertEqual(a.ndim, 2)
     # Property "shape"
     self.assertEqual(a.shape, (3, 5))
     # Property "strides"
     self.assertEqual(a.strides, (20, 4))
     # Property "dtype"
     self.assertEqual(a.dtype, ndt.int32)
コード例 #8
0
 def test_squeeze_errors(self):
     a = nd.zeros(1, 3, 1, 2, 1, ndt.int32)
     # Out of bound axis
     self.assertRaises(IndexError, nd.squeeze, a, axis=-6)
     self.assertRaises(IndexError, nd.squeeze, a, axis=5)
     self.assertRaises(IndexError, nd.squeeze, a, axis=(0, 5))
     self.assertRaises(IndexError, nd.squeeze, a, axis=(2, -6))
     # Dimension of non-one size
     self.assertRaises(IndexError, nd.squeeze, a, axis=1)
     self.assertRaises(IndexError, nd.squeeze, a, axis=(2,3))
     # Axis not an integer
     self.assertRaises(TypeError, nd.squeeze, a, axis=2.0)
     self.assertRaises(TypeError, nd.squeeze, a, axis=(0, 2.0+0j))
コード例 #9
0
 def test_squeeze_errors(self):
     a = nd.zeros(1, 3, 1, 2, 1, ndt.int32)
     # Out of bound axis
     self.assertRaises(IndexError, nd.squeeze, a, axis=-6)
     self.assertRaises(IndexError, nd.squeeze, a, axis=5)
     self.assertRaises(IndexError, nd.squeeze, a, axis=(0, 5))
     self.assertRaises(IndexError, nd.squeeze, a, axis=(2, -6))
     # Dimension of non-one size
     self.assertRaises(IndexError, nd.squeeze, a, axis=1)
     self.assertRaises(IndexError, nd.squeeze, a, axis=(2, 3))
     # Axis not an integer
     self.assertRaises(TypeError, nd.squeeze, a, axis=2.0)
     self.assertRaises(TypeError, nd.squeeze, a, axis=(0, 2.0 + 0j))
コード例 #10
0
 def test_squeeze_axis(self):
     a = nd.zeros(1, 3, 1, 2, 1, ndt.int32)
     self.assertEqual(a.shape, (1, 3, 1, 2, 1))
     self.assertEqual(nd.squeeze(a, axis=0).shape, (3, 1, 2, 1))
     self.assertEqual(nd.squeeze(a, axis=2).shape, (1, 3, 2, 1))
     self.assertEqual(nd.squeeze(a, axis=4).shape, (1, 3, 1, 2))
     self.assertEqual(nd.squeeze(a, axis=-5).shape, (3, 1, 2, 1))
     self.assertEqual(nd.squeeze(a, axis=-3).shape, (1, 3, 2, 1))
     self.assertEqual(nd.squeeze(a, axis=-1).shape, (1, 3, 1, 2))
     self.assertEqual(nd.squeeze(a, axis=(0,2)).shape, (3, 2, 1))
     self.assertEqual(nd.squeeze(a, axis=(0,4)).shape, (3, 1, 2))
     self.assertEqual(nd.squeeze(a, axis=(0,-1)).shape, (3, 1, 2))
     self.assertEqual(nd.squeeze(a, axis=(2,4)).shape, (1, 3, 2))
     self.assertEqual(nd.squeeze(a, axis=(0,2,4)).shape, (3, 2))
コード例 #11
0
 def test_squeeze_axis(self):
     a = nd.zeros(1, 3, 1, 2, 1, ndt.int32)
     self.assertEqual(a.shape, (1, 3, 1, 2, 1))
     self.assertEqual(nd.squeeze(a, axis=0).shape, (3, 1, 2, 1))
     self.assertEqual(nd.squeeze(a, axis=2).shape, (1, 3, 2, 1))
     self.assertEqual(nd.squeeze(a, axis=4).shape, (1, 3, 1, 2))
     self.assertEqual(nd.squeeze(a, axis=-5).shape, (3, 1, 2, 1))
     self.assertEqual(nd.squeeze(a, axis=-3).shape, (1, 3, 2, 1))
     self.assertEqual(nd.squeeze(a, axis=-1).shape, (1, 3, 1, 2))
     self.assertEqual(nd.squeeze(a, axis=(0, 2)).shape, (3, 2, 1))
     self.assertEqual(nd.squeeze(a, axis=(0, 4)).shape, (3, 1, 2))
     self.assertEqual(nd.squeeze(a, axis=(0, -1)).shape, (3, 1, 2))
     self.assertEqual(nd.squeeze(a, axis=(2, 4)).shape, (1, 3, 2))
     self.assertEqual(nd.squeeze(a, axis=(0, 2, 4)).shape, (3, 2))
コード例 #12
0
def zeros(dshape, ddesc=None):
    """Create an array and fill it with zeros.

    Parameters
    ----------
    dshape : datashape
        The datashape for the resulting array.

    ddesc : data descriptor instance
        This comes with the necessary info for storing the data.  If
        None, a DyND_DDesc will be used.

    Returns
    -------
    out : a concrete blaze array.

    """
    dshape = _normalize_dshape(dshape)

    if ddesc is None:
        ddesc = DyND_DDesc(nd.zeros(str(dshape), access='rw'))
        return Array(ddesc)
    if isinstance(ddesc, BLZ_DDesc):
        shape, dt = to_numpy(dshape)
        ddesc.blzarr = blz.zeros(shape,
                                 dt,
                                 rootdir=ddesc.path,
                                 mode=ddesc.mode,
                                 **ddesc.kwargs)
    elif isinstance(ddesc, HDF5_DDesc):
        obj = nd.as_numpy(nd.zeros(str(dshape)))
        with tb.open_file(ddesc.path, mode=ddesc.mode) as f:
            where, name = split_path(ddesc.datapath)
            f.create_earray(where, name, filters=ddesc.filters, obj=obj)
        ddesc.mode = 'a'  # change into 'a'ppend mode for further operations
    return Array(ddesc)
コード例 #13
0
 def test_access_zeros(self):
     a = nd.zeros(ndt.int32)
     self.assertEqual(a.access_flags, 'readwrite')
コード例 #14
0
 def test_access_zeros(self):
     a = nd.zeros(ndt.int32)
     self.assertEqual(a.access_flags, "immutable")
     a = nd.zeros(ndt.int32, access="rw")
     self.assertEqual(a.access_flags, "readwrite")
コード例 #15
0
 def test_access_zeros(self):
     a = nd.zeros(ndt.int32)
     self.assertEqual(a.access_flags, 'readwrite')