def do_test_sparse(self, dtype):
def count_elements(mat):
count = 0
for i in mat.indices():
count += 1
return count
size = 100
A1 = make_random_matrix(size, dtype, pyublas.SparseBuildMatrix)
A2 = pyublas.asarray(A1, dtype, pyublas.SparseExecuteMatrix)
self.assert_(count_elements(A1) == count_elements(A2))
import numpy import pyublas assert pyublas.have_sparse_wrappers() a = pyublas.zeros((5, 5), flavor=pyublas.SparseBuildMatrix, dtype=float) a[4, 2] = 19 b = numpy.random.randn(2, 2) a.add_block(2, 2, b) a_fast = pyublas.asarray(a, flavor=pyublas.SparseExecuteMatrix) vec = numpy.random.randn(5) res = a_fast * vec print a_fast print res
import numpy import pyublas assert pyublas.have_sparse_wrappers() a = pyublas.zeros((5,5), flavor=pyublas.SparseBuildMatrix, dtype=float) a[4,2] = 19 b = numpy.random.randn(2,2) a.add_block(2, 2, b) a_fast = pyublas.asarray(a, flavor=pyublas.SparseExecuteMatrix) vec = numpy.random.randn(5) res = a_fast * vec print a_fast print res
