Ejemplo n.º 1
0
 def test_broadcast_wrong_shapes(self):
     from _numpypy import zeros
     a = zeros((4, 3, 2))
     b = zeros((4, 2))
     exc = raises(ValueError, lambda: a + b)
     assert str(exc.value) == "operands could not be broadcast" \
         " together with shapes (4,3,2) (4,2)"
Ejemplo n.º 2
0
 def test_broadcast_wrong_shapes(self):
     from _numpypy import zeros
     a = zeros((4, 3, 2))
     b = zeros((4, 2))
     exc = raises(ValueError, lambda: a + b)
     assert str(exc.value) == "operands could not be broadcast" \
         " together with shapes (4,3,2) (4,2)"
Ejemplo n.º 3
0
    def test_str(self):
        from _numpypy import array, zeros
        a = array(range(5), float)
        assert str(a) == "[0.0 1.0 2.0 3.0 4.0]"
        assert str((2 * a)[:]) == "[0.0 2.0 4.0 6.0 8.0]"
        a = zeros(1001)
        assert str(a) == "[0.0 0.0 0.0 ..., 0.0 0.0 0.0]"

        a = array(range(5), dtype=long)
        assert str(a) == "[0 1 2 3 4]"
        a = array([True, False, True, False], dtype="?")
        assert str(a) == "[True False True False]"

        a = array(range(5), dtype="int8")
        assert str(a) == "[0 1 2 3 4]"

        a = array(range(5), dtype="int16")
        assert str(a) == "[0 1 2 3 4]"

        a = array((range(5), range(5, 10)), dtype="int16")
        assert str(a) == "[[0 1 2 3 4]\n [5 6 7 8 9]]"

        a = array(3, dtype=int)
        assert str(a) == "3"

        a = zeros((400, 400), dtype=int)
        assert str(a) == "[[0 0 0 ..., 0 0 0]\n [0 0 0 ..., 0 0 0]\n" \
           " [0 0 0 ..., 0 0 0]\n ...,\n [0 0 0 ..., 0 0 0]\n" \
           " [0 0 0 ..., 0 0 0]\n [0 0 0 ..., 0 0 0]]"
        a = zeros((2, 2, 2))
        r = str(a)
        assert r == '[[[0.0 0.0]\n  [0.0 0.0]]\n\n [[0.0 0.0]\n  [0.0 0.0]]]'
Ejemplo n.º 4
0
 def test_repr(self):
     from _numpypy import array, zeros
     int_size = array(5).dtype.itemsize
     a = array(range(5), float)
     assert repr(a) == "array([0.0, 1.0, 2.0, 3.0, 4.0])"
     a = array([], float)
     assert repr(a) == "array([], dtype=float64)"
     a = zeros(1001)
     assert repr(a) == "array([0.0, 0.0, 0.0, ..., 0.0, 0.0, 0.0])"
     a = array(range(5), long)
     if a.dtype.itemsize == int_size:
         assert repr(a) == "array([0, 1, 2, 3, 4])"
     else:
         assert repr(a) == "array([0, 1, 2, 3, 4], dtype=int64)"
     a = array(range(5), 'int32')
     if a.dtype.itemsize == int_size:
         assert repr(a) == "array([0, 1, 2, 3, 4])"
     else:
         assert repr(a) == "array([0, 1, 2, 3, 4], dtype=int32)"
     a = array([], long)
     assert repr(a) == "array([], dtype=int64)"
     a = array([True, False, True, False], "?")
     assert repr(a) == "array([True, False, True, False], dtype=bool)"
     a = zeros([])
     assert repr(a) == "array(0.0)"
     a = array(0.2)
     assert repr(a) == "array(0.2)"
Ejemplo n.º 5
0
    def test_str(self):
        from _numpypy import array, zeros
        a = array(range(5), float)
        assert str(a) == "[0.0 1.0 2.0 3.0 4.0]"
        assert str((2 * a)[:]) == "[0.0 2.0 4.0 6.0 8.0]"
        a = zeros(1001)
        assert str(a) == "[0.0 0.0 0.0 ..., 0.0 0.0 0.0]"

        a = array(range(5), dtype=long)
        assert str(a) == "[0 1 2 3 4]"
        a = array([True, False, True, False], dtype="?")
        assert str(a) == "[True False True False]"

        a = array(range(5), dtype="int8")
        assert str(a) == "[0 1 2 3 4]"

        a = array(range(5), dtype="int16")
        assert str(a) == "[0 1 2 3 4]"

        a = array((range(5), range(5, 10)), dtype="int16")
        assert str(a) == "[[0 1 2 3 4]\n [5 6 7 8 9]]"

        a = array(3, dtype=int)
        assert str(a) == "3"

        a = zeros((400, 400), dtype=int)
        assert str(a) == "[[0 0 0 ..., 0 0 0]\n [0 0 0 ..., 0 0 0]\n" \
           " [0 0 0 ..., 0 0 0]\n ...,\n [0 0 0 ..., 0 0 0]\n" \
           " [0 0 0 ..., 0 0 0]\n [0 0 0 ..., 0 0 0]]"
        a = zeros((2, 2, 2))
        r = str(a)
        assert r == '[[[0.0 0.0]\n  [0.0 0.0]]\n\n [[0.0 0.0]\n  [0.0 0.0]]]'
Ejemplo n.º 6
0
    def test_slice_reshape(self):
        from _numpypy import zeros, arange
        a = zeros((4, 2, 3))
        b = a[::2, :, :]
        b.shape = (2, 6)
        exc = raises(AttributeError, "b.shape = 12")
        assert str(exc.value) == \
                           "incompatible shape for a non-contiguous array"
        b = a[::2, :, :].reshape((2, 6))
        assert b.shape == (2, 6)
        b = arange(20)[1:17:2]
        b.shape = (4, 2)
        assert (b == [[1, 3], [5, 7], [9, 11], [13, 15]]).all()
        c = b.reshape((2, 4))
        assert (c == [[1, 3, 5, 7], [9, 11, 13, 15]]).all()

        z = arange(96).reshape((12, -1))
        assert z.shape == (12, 8)
        y = z.reshape((4, 3, 8))
        v = y[:, ::2, :]
        w = y.reshape(96)
        u = v.reshape(64)
        assert y[1, 2, 1] == z[5, 1]
        y[1, 2, 1] = 1000
        # z, y, w, v are views of eachother
        assert z[5, 1] == 1000
        assert v[1, 1, 1] == 1000
        assert w[41] == 1000
        # u is not a view, it is a copy!
        assert u[25] == 41

        a = zeros((5, 2))
        assert a.reshape(-1).shape == (10,)

        raises(ValueError, arange(10).reshape, (5, -1, -1))
Ejemplo n.º 7
0
 def test_repr(self):
     from _numpypy import array, zeros
     int_size = array(5).dtype.itemsize
     a = array(range(5), float)
     assert repr(a) == "array([0.0, 1.0, 2.0, 3.0, 4.0])"
     a = array([], float)
     assert repr(a) == "array([], dtype=float64)"
     a = zeros(1001)
     assert repr(a) == "array([0.0, 0.0, 0.0, ..., 0.0, 0.0, 0.0])"
     a = array(range(5), long)
     if a.dtype.itemsize == int_size:
         assert repr(a) == "array([0, 1, 2, 3, 4])"
     else:
         assert repr(a) == "array([0, 1, 2, 3, 4], dtype=int64)"
     a = array(range(5), 'int32')
     if a.dtype.itemsize == int_size:
         assert repr(a) == "array([0, 1, 2, 3, 4])"
     else:
         assert repr(a) == "array([0, 1, 2, 3, 4], dtype=int32)"
     a = array([], long)
     assert repr(a) == "array([], dtype=int64)"
     a = array([True, False, True, False], "?")
     assert repr(a) == "array([True, False, True, False], dtype=bool)"
     a = zeros([])
     assert repr(a) == "array(0.0)"
     a = array(0.2)
     assert repr(a) == "array(0.2)"
Ejemplo n.º 8
0
    def test_slice_reshape(self):
        from _numpypy import zeros, arange
        a = zeros((4, 2, 3))
        b = a[::2, :, :]
        b.shape = (2, 6)
        exc = raises(AttributeError, "b.shape = 12")
        assert str(exc.value) == \
                           "incompatible shape for a non-contiguous array"
        b = a[::2, :, :].reshape((2, 6))
        assert b.shape == (2, 6)
        b = arange(20)[1:17:2]
        b.shape = (4, 2)
        assert (b == [[1, 3], [5, 7], [9, 11], [13, 15]]).all()
        c = b.reshape((2, 4))
        assert (c == [[1, 3, 5, 7], [9, 11, 13, 15]]).all()

        z = arange(96).reshape((12, -1))
        assert z.shape == (12, 8)
        y = z.reshape((4, 3, 8))
        v = y[:, ::2, :]
        w = y.reshape(96)
        u = v.reshape(64)
        assert y[1, 2, 1] == z[5, 1]
        y[1, 2, 1] = 1000
        # z, y, w, v are views of eachother
        assert z[5, 1] == 1000
        assert v[1, 1, 1] == 1000
        assert w[41] == 1000
        # u is not a view, it is a copy!
        assert u[25] == 41

        a = zeros((5, 2))
        assert a.reshape(-1).shape == (10, )

        raises(ValueError, arange(10).reshape, (5, -1, -1))
Ejemplo n.º 9
0
 def test_shape(self):
     import _numpypy
     assert _numpypy.zeros(1).shape == (1,)
     assert _numpypy.zeros((2, 2)).shape == (2, 2)
     assert _numpypy.zeros((3, 1, 2)).shape == (3, 1, 2)
     assert _numpypy.array([[1], [2], [3]]).shape == (3, 1)
     assert len(_numpypy.zeros((3, 1, 2))) == 3
     raises(TypeError, len, _numpypy.zeros(()))
     raises(ValueError, _numpypy.array, [[1, 2], 3])
Ejemplo n.º 10
0
 def test_shape(self):
     import _numpypy
     assert _numpypy.zeros(1).shape == (1, )
     assert _numpypy.zeros((2, 2)).shape == (2, 2)
     assert _numpypy.zeros((3, 1, 2)).shape == (3, 1, 2)
     assert _numpypy.array([[1], [2], [3]]).shape == (3, 1)
     assert len(_numpypy.zeros((3, 1, 2))) == 3
     raises(TypeError, len, _numpypy.zeros(()))
     raises(ValueError, _numpypy.array, [[1, 2], 3])
Ejemplo n.º 11
0
 def test_multidim_setslice(self):
     from _numpypy import zeros, ones
     a = zeros((3, 3))
     b = ones((3, 3))
     a[:, 1:3] = b[:, 1:3]
     assert (a == [[0, 1, 1], [0, 1, 1], [0, 1, 1]]).all()
     a = zeros((3, 3))
     b = ones((3, 3))
     a[:, ::2] = b[:, ::2]
     assert (a == [[1, 0, 1], [1, 0, 1], [1, 0, 1]]).all()
Ejemplo n.º 12
0
 def test_multidim_setslice(self):
     from _numpypy import zeros, ones
     a = zeros((3, 3))
     b = ones((3, 3))
     a[:, 1:3] = b[:, 1:3]
     assert (a == [[0, 1, 1], [0, 1, 1], [0, 1, 1]]).all()
     a = zeros((3, 3))
     b = ones((3, 3))
     a[:, ::2] = b[:, ::2]
     assert (a == [[1, 0, 1], [1, 0, 1], [1, 0, 1]]).all()
Ejemplo n.º 13
0
 def test_broadcast_shape_agreement(self):
     from _numpypy import zeros, array
     a = zeros((3, 1, 3))
     b = array(((10, 11, 12), (20, 21, 22), (30, 31, 32)))
     c = ((a + b) == [b, b, b])
     assert c.all()
     a = array((((10, 11, 12), ), ((20, 21, 22), ), ((30, 31, 32), )))
     assert (a.shape == (3, 1, 3))
     d = zeros((3, 3))
     c = ((a + d) == [b, b, b])
     c = ((a + d) == array([[[10., 11., 12.]] * 3, [[20., 21., 22.]] * 3,
                            [[30., 31., 32.]] * 3]))
     assert c.all()
Ejemplo n.º 14
0
 def test_broadcast_shape_agreement(self):
     from _numpypy import zeros, array
     a = zeros((3, 1, 3))
     b = array(((10, 11, 12), (20, 21, 22), (30, 31, 32)))
     c = ((a + b) == [b, b, b])
     assert c.all()
     a = array((((10, 11, 12), ), ((20, 21, 22), ), ((30, 31, 32), )))
     assert(a.shape == (3, 1, 3))
     d = zeros((3, 3))
     c = ((a + d) == [b, b, b])
     c = ((a + d) == array([[[10., 11., 12.]] * 3,
                            [[20., 21., 22.]] * 3, [[30., 31., 32.]] * 3]))
     assert c.all()
Ejemplo n.º 15
0
    def test_zeros_long(self):
        from _numpypy import zeros, int64

        a = zeros(10, dtype=long)
        for i in range(10):
            assert isinstance(a[i], int64)
            assert a[1] == 0
Ejemplo n.º 16
0
 def test_broadcast_virtualview(self):
     from _numpypy import arange, zeros
     a = arange(8).reshape([2, 2, 2])
     b = (a + a)[1, 1]
     c = zeros((2, 2, 2))
     c[:] = b
     assert (c == [[[12, 14], [12, 14]], [[12, 14], [12, 14]]]).all()
Ejemplo n.º 17
0
 def test_slices(self):
     import _numpypy
     a = _numpypy.zeros((4, 3, 2))
     raises(IndexError, a.__getitem__, (4,))
     raises(IndexError, a.__getitem__, (3, 3))
     raises(IndexError, a.__getitem__, (slice(None), 3))
     a[0, 1, 1] = 13
     a[1, 2, 1] = 15
     b = a[0]
     assert len(b) == 3
     assert b.shape == (3, 2)
     assert b[1, 1] == 13
     b = a[1]
     assert b.shape == (3, 2)
     assert b[2, 1] == 15
     b = a[:, 1]
     assert b.shape == (4, 2)
     assert b[0, 1] == 13
     b = a[:, 1, :]
     assert b.shape == (4, 2)
     assert b[0, 1] == 13
     b = a[1, 2]
     assert b[1] == 15
     b = a[:]
     assert b.shape == (4, 3, 2)
     assert b[1, 2, 1] == 15
     assert b[0, 1, 1] == 13
     b = a[:][:, 1][:]
     assert b[2, 1] == 0.0
     assert b[0, 1] == 13
     raises(IndexError, b.__getitem__, (4, 1))
     assert a[0][1][1] == 13
     assert a[1][2][1] == 15
Ejemplo n.º 18
0
 def test_slices(self):
     import _numpypy
     a = _numpypy.zeros((4, 3, 2))
     raises(IndexError, a.__getitem__, (4, ))
     raises(IndexError, a.__getitem__, (3, 3))
     raises(IndexError, a.__getitem__, (slice(None), 3))
     a[0, 1, 1] = 13
     a[1, 2, 1] = 15
     b = a[0]
     assert len(b) == 3
     assert b.shape == (3, 2)
     assert b[1, 1] == 13
     b = a[1]
     assert b.shape == (3, 2)
     assert b[2, 1] == 15
     b = a[:, 1]
     assert b.shape == (4, 2)
     assert b[0, 1] == 13
     b = a[:, 1, :]
     assert b.shape == (4, 2)
     assert b[0, 1] == 13
     b = a[1, 2]
     assert b[1] == 15
     b = a[:]
     assert b.shape == (4, 3, 2)
     assert b[1, 2, 1] == 15
     assert b[0, 1, 1] == 13
     b = a[:][:, 1][:]
     assert b[2, 1] == 0.0
     assert b[0, 1] == 13
     raises(IndexError, b.__getitem__, (4, 1))
     assert a[0][1][1] == 13
     assert a[1][2][1] == 15
Ejemplo n.º 19
0
 def test_broadcast_virtualview(self):
     from _numpypy import arange, zeros
     a = arange(8).reshape([2, 2, 2])
     b = (a + a)[1, 1]
     c = zeros((2, 2, 2))
     c[:] = b
     assert (c == [[[12, 14], [12, 14]], [[12, 14], [12, 14]]]).all()
Ejemplo n.º 20
0
 def test_init(self):
     from _numpypy import zeros
     a = zeros(15)
     # Check that storage was actually zero'd.
     assert a[10] == 0.0
     # And check that changes stick.
     a[13] = 5.3
     assert a[13] == 5.3
Ejemplo n.º 21
0
 def test_any(self):
     from _numpypy import array, zeros
     a = array(range(5))
     assert a.any() == True
     b = zeros(5)
     assert b.any() == False
     c = array([])
     assert c.any() == False
Ejemplo n.º 22
0
 def test_init(self):
     from _numpypy import zeros
     a = zeros(15)
     # Check that storage was actually zero'd.
     assert a[10] == 0.0
     # And check that changes stick.
     a[13] = 5.3
     assert a[13] == 5.3
Ejemplo n.º 23
0
 def test_any(self):
     from _numpypy import array, zeros
     a = array(range(5))
     assert a.any() == True
     b = zeros(5)
     assert b.any() == False
     c = array([])
     assert c.any() == False
Ejemplo n.º 24
0
 def test_setslice_of_slice_array(self):
     from _numpypy import array, zeros
     a = zeros(5)
     a[::2] = array([9., 10., 11.])
     assert a[0] == 9.
     assert a[2] == 10.
     assert a[4] == 11.
     a[1:4:2][::-1] = array([1., 2.])
     assert a[0] == 9.
     assert a[1] == 2.
     assert a[2] == 10.
     assert a[3] == 1.
     assert a[4] == 11.
     a = zeros(10)
     a[::2][::-1][::2] = array(range(1, 4))
     assert a[8] == 1.
     assert a[4] == 2.
     assert a[0] == 3.
Ejemplo n.º 25
0
    def test_repr_multi(self):
        from _numpypy import arange, zeros
        a = zeros((3, 4))
        assert repr(a) == '''array([[0.0, 0.0, 0.0, 0.0],
       [0.0, 0.0, 0.0, 0.0],
       [0.0, 0.0, 0.0, 0.0]])'''
        a = zeros((2, 3, 4))
        assert repr(a) == '''array([[[0.0, 0.0, 0.0, 0.0],
        [0.0, 0.0, 0.0, 0.0],
        [0.0, 0.0, 0.0, 0.0]],

       [[0.0, 0.0, 0.0, 0.0],
        [0.0, 0.0, 0.0, 0.0],
        [0.0, 0.0, 0.0, 0.0]]])'''
        a = arange(1002).reshape((2, 501))
        assert repr(a) == '''array([[0, 1, 2, ..., 498, 499, 500],
       [501, 502, 503, ..., 999, 1000, 1001]])'''
        assert repr(a.T) == '''array([[0, 501],
Ejemplo n.º 26
0
    def test_repr_multi(self):
        from _numpypy import arange, zeros
        a = zeros((3, 4))
        assert repr(a) == '''array([[0.0, 0.0, 0.0, 0.0],
       [0.0, 0.0, 0.0, 0.0],
       [0.0, 0.0, 0.0, 0.0]])'''
        a = zeros((2, 3, 4))
        assert repr(a) == '''array([[[0.0, 0.0, 0.0, 0.0],
        [0.0, 0.0, 0.0, 0.0],
        [0.0, 0.0, 0.0, 0.0]],

       [[0.0, 0.0, 0.0, 0.0],
        [0.0, 0.0, 0.0, 0.0],
        [0.0, 0.0, 0.0, 0.0]]])'''
        a = arange(1002).reshape((2, 501))
        assert repr(a) == '''array([[0, 1, 2, ..., 498, 499, 500],
       [501, 502, 503, ..., 999, 1000, 1001]])'''
        assert repr(a.T) == '''array([[0, 501],
Ejemplo n.º 27
0
 def test_setslice_of_slice_array(self):
     from _numpypy import array, zeros
     a = zeros(5)
     a[::2] = array([9., 10., 11.])
     assert a[0] == 9.
     assert a[2] == 10.
     assert a[4] == 11.
     a[1:4:2][::-1] = array([1., 2.])
     assert a[0] == 9.
     assert a[1] == 2.
     assert a[2] == 10.
     assert a[3] == 1.
     assert a[4] == 11.
     a = zeros(10)
     a[::2][::-1][::2] = array(range(1, 4))
     assert a[8] == 1.
     assert a[4] == 2.
     assert a[0] == 3.
Ejemplo n.º 28
0
 def test_broadcast_scalar(self):
     from _numpypy import zeros
     a = zeros((4, 5), 'd')
     a[:, 1] = 3
     assert a[2, 1] == 3
     assert a[0, 2] == 0
     a[0, :] = 5
     assert a[0, 3] == 5
     assert a[2, 1] == 3
     assert a[3, 2] == 0
Ejemplo n.º 29
0
 def test_broadcast_scalar(self):
     from _numpypy import zeros
     a = zeros((4, 5), 'd')
     a[:, 1] = 3
     assert a[2, 1] == 3
     assert a[0, 2] == 0
     a[0, :] = 5
     assert a[0, 3] == 5
     assert a[2, 1] == 3
     assert a[3, 2] == 0
Ejemplo n.º 30
0
 def test_getsetitem(self):
     import _numpypy
     a = _numpypy.zeros((2, 3, 1))
     raises(IndexError, a.__getitem__, (2, 0, 0))
     raises(IndexError, a.__getitem__, (0, 3, 0))
     raises(IndexError, a.__getitem__, (0, 0, 1))
     assert a[1, 1, 0] == 0
     a[1, 2, 0] = 3
     assert a[1, 2, 0] == 3
     assert a[1, 1, 0] == 0
     assert a[1, -1, 0] == 3
Ejemplo n.º 31
0
 def test_getsetitem(self):
     import _numpypy
     a = _numpypy.zeros((2, 3, 1))
     raises(IndexError, a.__getitem__, (2, 0, 0))
     raises(IndexError, a.__getitem__, (0, 3, 0))
     raises(IndexError, a.__getitem__, (0, 0, 1))
     assert a[1, 1, 0] == 0
     a[1, 2, 0] = 3
     assert a[1, 2, 0] == 3
     assert a[1, 1, 0] == 0
     assert a[1, -1, 0] == 3
Ejemplo n.º 32
0
 def test_reshape(self):
     from _numpypy import array, zeros
     a = array(range(12))
     exc = raises(ValueError, "b = a.reshape((3, 10))")
     assert str(exc.value) == "total size of new array must be unchanged"
     b = a.reshape((3, 4))
     assert b.shape == (3, 4)
     assert (b == [range(4), range(4, 8), range(8, 12)]).all()
     b[:, 0] = 1000
     assert (a == [1000, 1, 2, 3, 1000, 5, 6, 7, 1000, 9, 10, 11]).all()
     a = zeros((4, 2, 3))
     a.shape = (12, 2)
Ejemplo n.º 33
0
 def test_reshape(self):
     from _numpypy import array, zeros
     a = array(range(12))
     exc = raises(ValueError, "b = a.reshape((3, 10))")
     assert str(exc.value) == "total size of new array must be unchanged"
     b = a.reshape((3, 4))
     assert b.shape == (3, 4)
     assert (b == [range(4), range(4, 8), range(8, 12)]).all()
     b[:, 0] = 1000
     assert (a == [1000, 1, 2, 3, 1000, 5, 6, 7, 1000, 9, 10, 11]).all()
     a = zeros((4, 2, 3))
     a.shape = (12, 2)
Ejemplo n.º 34
0
 def test_str_slice(self):
     from _numpypy import array, zeros
     a = array(range(5), float)
     b = a[1::2]
     assert str(b) == "[1.0 3.0]"
     a = zeros(2002)
     b = a[::2]
     assert str(b) == "[0.0 0.0 0.0 ..., 0.0 0.0 0.0]"
     a = array((range(5), range(5, 10)), dtype="int16")
     b = a[1, 2:]
     assert str(b) == "[7 8 9]"
     b = a[2:1, ]
     assert str(b) == "[]"
Ejemplo n.º 35
0
 def test_str_slice(self):
     from _numpypy import array, zeros
     a = array(range(5), float)
     b = a[1::2]
     assert str(b) == "[1.0 3.0]"
     a = zeros(2002)
     b = a[::2]
     assert str(b) == "[0.0 0.0 0.0 ..., 0.0 0.0 0.0]"
     a = array((range(5), range(5, 10)), dtype="int16")
     b = a[1, 2:]
     assert str(b) == "[7 8 9]"
     b = a[2:1, ]
     assert str(b) == "[]"
Ejemplo n.º 36
0
 def test_repr_slice(self):
     from _numpypy import array, zeros
     a = array(range(5), float)
     b = a[1::2]
     assert repr(b) == "array([1.0, 3.0])"
     a = zeros(2002)
     b = a[::2]
     assert repr(b) == "array([0.0, 0.0, 0.0, ..., 0.0, 0.0, 0.0])"
     a = array((range(5), range(5, 10)), dtype="int16")
     b = a[1, 2:]
     assert repr(b) == "array([7, 8, 9], dtype=int16)"
     # an empty slice prints its shape
     b = a[2:1, ]
     assert repr(b) == "array([], shape=(0, 5), dtype=int16)"
Ejemplo n.º 37
0
 def test_setitem_slice(self):
     import _numpypy
     a = _numpypy.zeros((3, 4))
     a[1] = [1, 2, 3, 4]
     assert a[1, 2] == 3
     raises(TypeError, a[1].__setitem__, [1, 2, 3])
     a = _numpypy.array([[1, 2], [3, 4]])
     assert (a == [[1, 2], [3, 4]]).all()
     a[1] = _numpypy.array([5, 6])
     assert (a == [[1, 2], [5, 6]]).all()
     a[:, 1] = _numpypy.array([8, 10])
     assert (a == [[1, 8], [5, 10]]).all()
     a[0, ::-1] = _numpypy.array([11, 12])
     assert (a == [[12, 11], [5, 10]]).all()
Ejemplo n.º 38
0
    def test_debug_repr(self):
        from _numpypy import zeros, sin
        from _numpypy.pypy import debug_repr
        a = zeros(1)
        assert debug_repr(a) == 'Array'
        assert debug_repr(a + a) == 'Call2(add, Array, Array)'
        assert debug_repr(a[::2]) == 'Slice'
        assert debug_repr(a + 2) == 'Call2(add, Array, Scalar)'
        assert debug_repr(a + a.flat) == 'Call2(add, Array, Slice)'
        assert debug_repr(sin(a)) == 'Call1(sin, Array)'

        b = a + a
        b[0] = 3
        assert debug_repr(b) == 'Array'
Ejemplo n.º 39
0
 def test_setitem_slice(self):
     import _numpypy
     a = _numpypy.zeros((3, 4))
     a[1] = [1, 2, 3, 4]
     assert a[1, 2] == 3
     raises(TypeError, a[1].__setitem__, [1, 2, 3])
     a = _numpypy.array([[1, 2], [3, 4]])
     assert (a == [[1, 2], [3, 4]]).all()
     a[1] = _numpypy.array([5, 6])
     assert (a == [[1, 2], [5, 6]]).all()
     a[:, 1] = _numpypy.array([8, 10])
     assert (a == [[1, 8], [5, 10]]).all()
     a[0, :: -1] = _numpypy.array([11, 12])
     assert (a == [[12, 11], [5, 10]]).all()
Ejemplo n.º 40
0
    def test_debug_repr(self):
        from _numpypy import zeros, sin
        from _numpypy.pypy import debug_repr
        a = zeros(1)
        assert debug_repr(a) == 'Array'
        assert debug_repr(a + a) == 'Call2(add, Array, Array)'
        assert debug_repr(a[::2]) == 'Slice'
        assert debug_repr(a + 2) == 'Call2(add, Array, Scalar)'
        assert debug_repr(a + a.flat) == 'Call2(add, Array, Slice)'
        assert debug_repr(sin(a)) == 'Call1(sin, Array)'

        b = a + a
        b[0] = 3
        assert debug_repr(b) == 'Array'
Ejemplo n.º 41
0
 def test_repr_slice(self):
     from _numpypy import array, zeros
     a = array(range(5), float)
     b = a[1::2]
     assert repr(b) == "array([1.0, 3.0])"
     a = zeros(2002)
     b = a[::2]
     assert repr(b) == "array([0.0, 0.0, 0.0, ..., 0.0, 0.0, 0.0])"
     a = array((range(5), range(5, 10)), dtype="int16")
     b = a[1, 2:]
     assert repr(b) == "array([7, 8, 9], dtype=int16)"
     # an empty slice prints its shape
     b = a[2:1, ]
     assert repr(b) == "array([], shape=(0, 5), dtype=int16)"
Ejemplo n.º 42
0
def arange(start, stop=None, step=1, dtype=None):
    '''arange([start], stop[, step], dtype=None)
    Generate values in the half-interval [start, stop).
    '''
    if stop is None:
        stop = start
        start = 0
    if dtype is None:
        test = _numpypy.array([start, stop, step, 0])
        dtype = test.dtype
    arr = _numpypy.zeros(int(math.ceil((stop - start) / step)), dtype=dtype)
    i = start
    for j in range(arr.size):
        arr[j] = i
        i += step
    return arr
Ejemplo n.º 43
0
def arange(start, stop=None, step=1, dtype=None):
    '''arange([start], stop[, step], dtype=None)
    Generate values in the half-interval [start, stop).
    '''
    if stop is None:
        stop = start
        start = 0
    if dtype is None:
        test = _numpypy.array([start, stop, step, 0])
        dtype = test.dtype
    arr = _numpypy.zeros(int(math.ceil((stop - start) / step)), dtype=dtype)
    i = start
    for j in range(arr.size):
        arr[j] = i
        i += step
    return arr
Ejemplo n.º 44
0
def eye(n, m=None, k=0, dtype=None):
    if m is None:
        m = n
    a = _numpypy.zeros((n, m), dtype=dtype)
    ni = 0
    mi = 0

    if k < 0:
        p = n + k
        ni = -k
    else:
        p = n - k
        mi = k

    while ni < n and mi < m:
        a[ni][mi] = 1
        ni += 1
        mi += 1
    return a
Ejemplo n.º 45
0
def identity(n, dtype=None):
    a = _numpypy.zeros((n,n), dtype=dtype)
    for i in range(n):
        a[i][i] = 1
    return a
Ejemplo n.º 46
0
 def test_init(self):
     import _numpypy
     a = _numpypy.zeros((2, 2))
     assert len(a) == 2
Ejemplo n.º 47
0
 def test_broadcast_setslice(self):
     from _numpypy import zeros, ones
     a = zeros((10, 10))
     b = ones(10)
     a[:, :] = b
     assert a[3, 5] == 1
Ejemplo n.º 48
0
 def test_broadcast_call2(self):
     from _numpypy import zeros, ones
     a = zeros((4, 1, 5))
     b = ones((4, 3, 5))
     b[:] = (a + a)
     assert (b == zeros((4, 3, 5))).all()
Ejemplo n.º 49
0
 def test_slice_copy(self):
     from _numpypy import zeros
     a = zeros((10, 10))
     b = a[0].copy()
     assert (b == zeros(10)).all()
Ejemplo n.º 50
0
 def test_where(self):
     from _numpypy import where, ones, zeros, array
     a = [1, 2, 3, 0, -3]
     a = where(array(a) > 0, ones(5), zeros(5))
     assert (a == [1, 1, 1, 0, 0]).all()
Ejemplo n.º 51
0
 def test_slice_copy(self):
     from _numpypy import zeros
     a = zeros((10, 10))
     b = a[0].copy()
     assert (b == zeros(10)).all()
Ejemplo n.º 52
0
 def test_init(self):
     import _numpypy
     a = _numpypy.zeros((2, 2))
     assert len(a) == 2
Ejemplo n.º 53
0
 def test_broadcast_setslice(self):
     from _numpypy import zeros, ones
     a = zeros((10, 10))
     b = ones(10)
     a[:, :] = b
     assert a[3, 5] == 1
Ejemplo n.º 54
0
 def test_broadcast_call2(self):
     from _numpypy import zeros, ones
     a = zeros((4, 1, 5))
     b = ones((4, 3, 5))
     b[:] = (a + a)
     assert (b == zeros((4, 3, 5))).all()
Ejemplo n.º 55
0
    def test_zeros_bool(self):
        from _numpypy import zeros, False_

        a = zeros(10, dtype=bool)
        for i in range(10):
            assert a[i] is False_
Ejemplo n.º 56
0
 def test_where(self):
     from _numpypy import where, ones, zeros, array
     a = [1, 2, 3, 0, -3]
     a = where(array(a) > 0, ones(5), zeros(5))
     assert (a == [1, 1, 1, 0, 0]).all()
Ejemplo n.º 57
0
def identity(n, dtype=None):
    a = _numpypy.zeros((n, n), dtype=dtype)
    for i in range(n):
        a[i][i] = 1
    return a
Ejemplo n.º 58
0
    def test_zeros_bool(self):
        from _numpypy import zeros, False_

        a = zeros(10, dtype=bool)
        for i in range(10):
            assert a[i] is False_
Ejemplo n.º 59
0
 def test_zeros_long(self):
     from _numpypy import zeros, longlong
     a = zeros(10, dtype=long)
     for i in range(10):
         assert isinstance(a[i], longlong)
         assert a[1] == 0
Ejemplo n.º 60
0
 def test_where_invalidates(self):
     from _numpypy import where, ones, zeros, array
     a = array([1, 2, 3, 0, -3])
     b = where(a > 0, ones(5), zeros(5))
     a[0] = 0
     assert (b == [1, 1, 1, 0, 0]).all()