Beispiel #1
0
 def test_dense(self):
     sizes = [(2,2),(3,3),(5,1),(1,5)]
     sizes += [(2,2,2),(4,3,2),(1,1,5),(1,5,1),(5,1,1)]
     for dims in sizes:
         mats = [arange(prod(dims)).reshape(dims),rand(*dims)]    
         for A in mats:
             formats = ['binary','ascii']
             if rank(A) <= 2: formats.append('basic') #use basic when possible
             for format in formats:
                 write_array(filename,A,format=format)
                 
                 B = read_array(filename)
                 assert_almost_equal(A,B,decimal=12)
Beispiel #2
0
    def test_dense(self):
        sizes = [(2, 2), (3, 3), (5, 1), (1, 5)]
        sizes += [(2, 2, 2), (4, 3, 2), (1, 1, 5), (1, 5, 1), (5, 1, 1)]
        for dims in sizes:
            mats = [arange(prod(dims)).reshape(dims), rand(*dims)]
            for A in mats:
                formats = ['binary', 'ascii']
                if rank(A) <= 2:
                    formats.append('basic')  #use basic when possible
                for format in formats:
                    write_array(filename, A, format=format)

                    B = read_array(filename)
                    assert_almost_equal(A, B, decimal=12)
Beispiel #3
0
    def test_sparse(self):
        sizes = [(2, 2), (3, 3), (1, 10), (10, 1), (10, 10)]
        for dims in sizes:
            base_mats = []
            base_mats.append(
                (rand(*dims) < 0.5) * rand(*dims))  #random matrix with 50% nnz
            base_mats.append(zeros(dims))  #empty matrix
            base_mats.append(arange(prod(dims)).reshape(dims))

            mats = []
            for base_mat in base_mats:
                mats.append(csr_matrix(base_mat))
                mats.append(csc_matrix(base_mat))
                mats.append(coo_matrix(base_mat))

            for A in mats:
                formats = ['binary', 'ascii']
                for format in formats:
                    write_array(filename, A, format=format)

                    B = read_array(filename)
                    assert_almost_equal(A.todense(), B.todense(), decimal=12)
                    assert_equal(type(A), type(B))
Beispiel #4
0
    def test_sparse(self):
        sizes = [(2,2),(3,3),(1,10),(10,1),(10,10)]
        for dims in sizes:
            base_mats = []
            base_mats.append((rand(*dims) < 0.5)*rand(*dims))  #random matrix with 50% nnz
            base_mats.append(zeros(dims))                      #empty matrix
            base_mats.append(arange(prod(dims)).reshape(dims))

            mats = []
            for base_mat in base_mats:
                mats.append(csr_matrix(base_mat))
                mats.append(csc_matrix(base_mat))
                mats.append(coo_matrix(base_mat))
            
            
            for A in mats:
                formats = ['binary','ascii']                        
                for format in formats:
                    write_array(filename,A,format=format)
                    
                    B = read_array(filename)
                    assert_almost_equal(A.todense(),B.todense(),decimal=12)
                    assert_equal(type(A),type(B))