Application: gvSIG desktop » gvSIG 3D

import numpy as np

from numpy import ( array, ones, r_, mgrid, unravel_index, zeros, where,

                    ndenumerate, fill_diagonal, diag_indices,

                    diag_indices_from )

    def test_linspace_equivalence(self):

        y,st = np.linspace(2,10,retstep=1)

        assert_almost_equal(st,8/49.0)

        assert_array_almost_equal(y,mgrid[2:10:50j],13)

class TestNdenumerate(TestCase):

    def test_basic(self):

        a = array([[1,2], [3,4]])

        assert_equal(list(ndenumerate(a)),

                     [((0,0), 1), ((0,1), 2), ((1,0), 3), ((1,1), 4)])

def test_fill_diagonal():

    a = zeros((3, 3),int)

    fill_diagonal(a, 5)

    yield (assert_array_equal, a,

           array([[5, 0, 0],

                  [0, 5, 0],

                  [0, 0, 5]]))

    # The same function can operate on a 4-d array:

    a = zeros((3, 3, 3, 3), int)

    fill_diagonal(a, 4)

    i = array([0, 1, 2])

    yield (assert_equal, where(a != 0), (i, i, i, i))

def test_diag_indices():

    di = diag_indices(4)

    a = array([[1, 2, 3, 4],

               [5, 6, 7, 8],

               [9, 10, 11, 12],

               [13, 14, 15, 16]])

    a[di] = 100

    yield (assert_array_equal, a,

           array([[100,   2,   3,   4],

                  [  5, 100,   7,   8],

                  [  9,  10, 100,  12],

                  [ 13,  14,  15, 100]]))

    # Now, we create indices to manipulate a 3-d array:

    d3 = diag_indices(2, 3)

    # And use it to set the diagonal of a zeros array to 1:

    a = zeros((2, 2, 2),int)

    a[d3] = 1

    yield (assert_array_equal, a,

           array([[[1, 0],

                   [0, 0]],

                  [[0, 0],

                   [0, 1]]]) )

def test_diag_indices_from():

    x = np.random.random((4, 4))

    r, c = diag_indices_from(x)

    assert_array_equal(r, np.arange(4))

    assert_array_equal(c, np.arange(4))