Exemple #1
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))
Exemple #2
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))
Exemple #3
0
    def test_reduceND(self):
        from _numpypy import add, arange

        a = arange(12).reshape(3, 4)
        assert (add.reduce(a, 0) == [12, 15, 18, 21]).all()
        assert (add.reduce(a, 1) == [6.0, 22.0, 38.0]).all()
        raises(ValueError, add.reduce, a, 2)
Exemple #4
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()
Exemple #5
0
    def test_count_reduce_items(self):
        from _numpypy import count_reduce_items, arange

        a = arange(24).reshape(2, 3, 4)
        assert count_reduce_items(a) == 24
        assert count_reduce_items(a, 1) == 3
        assert count_reduce_items(a, (1, 2)) == 3 * 4
Exemple #6
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()
Exemple #7
0
    def test_count_nonzero(self):
        from _numpypy import where, count_nonzero, arange

        a = arange(10)
        assert count_nonzero(a) == 9
        a[9] = 0
        assert count_nonzero(a) == 8
Exemple #8
0
    def test_reduce_keepdims(self):
        from _numpypy import add, arange

        a = arange(12).reshape(3, 4)
        b = add.reduce(a, 0, keepdims=True)
        assert b.shape == (1, 4)
        assert (add.reduce(a, 0, keepdims=True) == [12, 15, 18, 21]).all()
Exemple #9
0
 def test_virtual_views(self):
     from _numpypy import arange
     a = arange(15)
     c = (a + a)
     d = c[::2]
     assert d[3] == 12
     c[6] = 5
     assert d[3] == 5
     a = arange(15)
     c = (a + a)
     d = c[::2][::2]
     assert d[1] == 8
     b = a + a
     c = b[::2]
     c[:] = 3
     assert b[0] == 3
     assert b[1] == 2
Exemple #10
0
 def test_virtual_views(self):
     from _numpypy import arange
     a = arange(15)
     c = (a + a)
     d = c[::2]
     assert d[3] == 12
     c[6] = 5
     assert d[3] == 5
     a = arange(15)
     c = (a + a)
     d = c[::2][::2]
     assert d[1] == 8
     b = a + a
     c = b[::2]
     c[:] = 3
     assert b[0] == 3
     assert b[1] == 2
Exemple #11
0
 def test_bitwise(self):
     from _numpypy import bitwise_and, bitwise_or, bitwise_xor, arange, array
     a = arange(6).reshape(2, 3)
     assert (a & 1 == [[0, 1, 0], [1, 0, 1]]).all()
     assert (a & 1 == bitwise_and(a, 1)).all()
     assert (a | 1 == [[1, 1, 3], [3, 5, 5]]).all()
     assert (a | 1 == bitwise_or(a, 1)).all()
     assert (a ^ 3 == bitwise_xor(a, 3)).all()
     raises(TypeError, 'array([1.0]) & 1')
Exemple #12
0
    def test_count_reduce_items(self):
        from _numpypy import count_reduce_items, arange

        a = arange(24).reshape(2, 3, 4)
        assert count_reduce_items(a) == 24
        assert count_reduce_items(a, 1) == 3
        assert count_reduce_items(a, (1, 2)) == 3 * 4
        raises(ValueError, count_reduce_items, a, -4)
        raises(ValueError, count_reduce_items, a, (0, 2, -4))
Exemple #13
0
 def test_bitwise(self):
     from _numpypy import bitwise_and, bitwise_or, bitwise_xor, arange, array
     a = arange(6).reshape(2, 3)
     assert (a & 1 == [[0, 1, 0], [1, 0, 1]]).all()
     assert (a & 1 == bitwise_and(a, 1)).all()
     assert (a | 1 == [[1, 1, 3], [3, 5, 5]]).all()
     assert (a | 1 == bitwise_or(a, 1)).all()
     assert (a ^ 3 == bitwise_xor(a, 3)).all()
     raises(TypeError, 'array([1.0]) & 1')
Exemple #14
0
    def test_copy(self):
        from _numpypy import arange, array
        a = arange(5)
        b = a.copy()
        for i in xrange(5):
            assert b[i] == a[i]
        a[3] = 22
        assert b[3] == 3

        a = array(1)
        assert a.copy() == a

        a = arange(8)
        b = a[::2]
        c = b.copy()
        assert (c == b).all()

        a = arange(15).reshape(5,3)
        b = a.copy()
        assert (b == a).all()
Exemple #15
0
    def test_copy(self):
        from _numpypy import arange, array
        a = arange(5)
        b = a.copy()
        for i in xrange(5):
            assert b[i] == a[i]
        a[3] = 22
        assert b[3] == 3

        a = array(1)
        assert a.copy() == a

        a = arange(8)
        b = a[::2]
        c = b.copy()
        assert (c == b).all()

        a = arange(15).reshape(5, 3)
        b = a.copy()
        assert (b == a).all()
Exemple #16
0
 def test_app_reshape(self):
     from _numpypy import arange, array, dtype, reshape
     a = arange(12)
     b = reshape(a, (3, 4))
     assert b.shape == (3, 4)
     a = range(12)
     b = reshape(a, (3, 4))
     assert b.shape == (3, 4)
     a = array(range(105)).reshape(3, 5, 7)
     assert a.reshape(1, -1).shape == (1, 105)
     assert a.reshape(1, 1, -1).shape == (1, 1, 105)
     assert a.reshape(-1, 1, 1).shape == (105, 1, 1)
Exemple #17
0
 def test_app_reshape(self):
     from _numpypy import arange, array, dtype, reshape
     a = arange(12)
     b = reshape(a, (3, 4))
     assert b.shape == (3, 4)
     a = range(12)
     b = reshape(a, (3, 4))
     assert b.shape == (3, 4)
     a = array(range(105)).reshape(3, 5, 7)
     assert a.reshape(1, -1).shape == (1, 105)
     assert a.reshape(1, 1, -1).shape == (1, 1, 105)
     assert a.reshape(-1, 1, 1).shape == (105, 1, 1)
Exemple #18
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],
Exemple #19
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],
Exemple #20
0
 def test_arange(self):
     from _numpypy import arange, array, dtype
     a = arange(3)
     assert (a == [0, 1, 2]).all()
     assert a.dtype is dtype(int)
     a = arange(3.0)
     assert (a == [0., 1., 2.]).all()
     assert a.dtype is dtype(float)
     a = arange(3, 7)
     assert (a == [3, 4, 5, 6]).all()
     assert a.dtype is dtype(int)
     a = arange(3, 7, 2)
     assert (a == [3, 5]).all()
     a = arange(3, dtype=float)
     assert (a == [0., 1., 2.]).all()
     assert a.dtype is dtype(float)
     a = arange(0, 0.8, 0.1)
     assert len(a) == 8
     assert arange(False, True, True).dtype is dtype(int)
Exemple #21
0
 def test_arange(self):
     from _numpypy import arange, array, dtype
     a = arange(3)
     assert (a == [0, 1, 2]).all()
     assert a.dtype is dtype(int)
     a = arange(3.0)
     assert (a == [0., 1., 2.]).all()
     assert a.dtype is dtype(float)
     a = arange(3, 7)
     assert (a == [3, 4, 5, 6]).all()
     assert a.dtype is dtype(int)
     a = arange(3, 7, 2)
     assert (a == [3, 5]).all()
     a = arange(3, dtype=float)
     assert (a == [0., 1., 2.]).all()
     assert a.dtype is dtype(float)
     a = arange(0, 0.8, 0.1)
     assert len(a) == 8
     assert arange(False, True, True).dtype is dtype(int)
Exemple #22
0
 def test_reduce_nd(self):
     from numpypy import arange, array, multiply
     a = arange(15).reshape(5, 3)
     assert a.sum() == 105
     assert a.max() == 14
     assert array([]).sum() == 0.0
     raises(ValueError, 'array([]).max()')
     assert (a.sum(0) == [30, 35, 40]).all()
     assert (a.sum(axis=0) == [30, 35, 40]).all()
     assert (a.sum(1) == [3, 12, 21, 30, 39]).all()
     assert (a.max(0) == [12, 13, 14]).all()
     assert (a.max(1) == [2, 5, 8, 11, 14]).all()
     assert ((a + a).max() == 28)
     assert ((a + a).max(0) == [24, 26, 28]).all()
     assert ((a + a).sum(1) == [6, 24, 42, 60, 78]).all()
     assert (multiply.reduce(a) == array([0, 3640, 12320])).all()
     a = array(range(105)).reshape(3, 5, 7)
     assert (a[:, 1, :].sum(0) == [126, 129, 132, 135, 138, 141, 144]).all()
     assert (a[:, 1, :].sum(1) == [70, 315, 560]).all()
     raises(ValueError, 'a[:, 1, :].sum(2)')
     assert ((a + a).T.sum(2).T == (a + a).sum(0)).all()
     assert (a.reshape(1, -1).sum(0) == range(105)).all()
     assert (a.reshape(1, -1).sum(1) == 5460)
Exemple #23
0
 def test_reduce_nd(self):
     from numpypy import arange, array, multiply
     a = arange(15).reshape(5, 3)
     assert a.sum() == 105
     assert a.max() == 14
     assert array([]).sum() == 0.0
     raises(ValueError, 'array([]).max()')
     assert (a.sum(0) == [30, 35, 40]).all()
     assert (a.sum(axis=0) == [30, 35, 40]).all()
     assert (a.sum(1) == [3, 12, 21, 30, 39]).all()
     assert (a.max(0) == [12, 13, 14]).all()
     assert (a.max(1) == [2, 5, 8, 11, 14]).all()
     assert ((a + a).max() == 28)
     assert ((a + a).max(0) == [24, 26, 28]).all()
     assert ((a + a).sum(1) == [6, 24, 42, 60, 78]).all()
     assert (multiply.reduce(a) == array([0, 3640, 12320])).all()
     a = array(range(105)).reshape(3, 5, 7)
     assert (a[:, 1, :].sum(0) == [126, 129, 132, 135, 138, 141, 144]).all()
     assert (a[:, 1, :].sum(1) == [70, 315, 560]).all()
     raises (ValueError, 'a[:, 1, :].sum(2)')
     assert ((a + a).T.sum(2).T == (a + a).sum(0)).all()
     assert (a.reshape(1,-1).sum(0) == range(105)).all()
     assert (a.reshape(1,-1).sum(1) == 5460)
Exemple #24
0
 def test_true_divide(self):
     from _numpypy import arange, array, true_divide
     assert (true_divide(arange(3), array([2, 2, 2])) == array([0, 0.5, 1])).all()
Exemple #25
0
 def test_true_divide(self):
     from _numpypy import arange, array, true_divide
     assert (true_divide(arange(3), array([2, 2, 2])) == array([0, 0.5,
                                                                1])).all()
Exemple #26
0
 def test_count_reduce_items(self):
     from _numpypy import count_reduce_items, arange
     a = arange(24).reshape(2, 3, 4)
     assert count_reduce_items(a) == 24
     assert count_reduce_items(a, 1) == 3
     assert count_reduce_items(a, (1, 2)) == 3 * 4
Exemple #27
0
 def test_reshape_varargs(self):
     from _numpypy import arange
     z = arange(96).reshape(12, -1)
     y = z.reshape(4, 3, 8)
     assert y.shape == (4, 3, 8)
Exemple #28
0
 def test_reduce_keepdims(self):
     from _numpypy import add, arange
     a = arange(12).reshape(3, 4)
     b = add.reduce(a, 0, keepdims=True)
     assert b.shape == (1, 4)
     assert (add.reduce(a, 0, keepdims=True) == [12, 15, 18, 21]).all()
Exemple #29
0
 def test_reduceND(self):
     from _numpypy import add, arange
     a = arange(12).reshape(3, 4)
     assert (add.reduce(a, 0) == [12, 15, 18, 21]).all()
     assert (add.reduce(a, 1) == [6.0, 22.0, 38.0]).all()
Exemple #30
0
 def test_reshape_varargs(self):
     from _numpypy import arange
     z = arange(96).reshape(12, -1)
     y = z.reshape(4, 3, 8)
     assert y.shape == (4, 3, 8)