1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
|
#!/usr/bin/env python
#
# Created by: Travis Oliphant, April 2004
#
""" Test functions for Sparse matrices
"""
__usage__ = """
Build sparse:
python setup_sparse.py build
Run tests if scipy is installed:
python -c 'import scipy;scipy.sparse.test(<level>)'
Run tests if sparse is not installed:
python tests/test_Sparse.py [<level>]
"""
import scipy_base
from scipy_base import arange, zeros, array, dot
import sys
from scipy_test.testing import *
set_package_path()
from sparse import csc_matrix, csr_matrix
restore_path()
class _test_cs(ScipyTestCase):
def setUp(self):
self.dat = array([[1,0,0,2],[3,0,1,0],[0,2,0,0]],'d')
self.datsp = self.spmatrix(self.dat)
def check_getelement(self):
assert_equal(self.datsp[0,0],1)
assert_equal(self.datsp[0,1],0)
assert_equal(self.datsp[1,0],3)
assert_equal(self.datsp[2,1],2)
def check_todense(self):
chk = self.datsp.todense()
assert_array_equal(chk,self.dat)
def check_setelement(self):
a = self.datsp - self.datsp
a[1,2] = 4.0
a[0,1] = 3
a[2,0] = 2.0
assert_array_equal(a.todense(),[[0,3,0,0],[0,0,4,0],[2,0,0,0]])
def check_add(self):
a = self.datsp
b = self.datsp.copy()
b[0,2] = 2.0
c = a + b
assert_array_equal(c.todense(),[[2,0,2,4],[6,0,2,0],[0,4,0,0]])
def check_elmul(self):
a = self.datsp
b = self.datsp.copy()
b[0,2] = 2.0
c = a ** b
assert_array_equal(c.todense(),[[1,0,0,4],[9,0,1,0],[0,4,0,0]])
def check_matvec(self):
b = self.spmatrix(array([[3,0,0],[0,1,0],[2,0,3.0],[2,3,0]]))
assert_array_almost_equal(b*[1,2,3],dot(b.todense(),[1,2,3]))
assert_array_almost_equal([1,2,3,4]*b,dot([1,2,3,4],b.todense()))
def check_matmat(self):
asp = self.spmatrix(array([[3,0,0],[0,1,0],[2,0,3.0],[2,3,0]]))
bsp = self.spmatrix(array([[0,1],[1,0],[0,2]],'d'))
assert_array_almost_equal((asp*bsp).todense(),dot(asp.todense(),bsp.todense()))
def check_tocoo(self):
a = self.datsp.tocoo()
assert_array_almost_equal(a.todense(),self.dat)
def check_tocsc(self):
a = self.datsp.tocsc()
assert_array_almost_equal(a.todense(),self.dat)
def check_tocsr(self):
a = self.datsp.tocsr()
assert_array_almost_equal(a.todense(),self.dat)
class test_csr(_test_cs):
spmatrix = csr_matrix
def check_constructor1(self):
b = array([[0,4,0],
[3,0,1],
[0,2,0]],'d')
bsp = csr_matrix(b)
assert_array_almost_equal(bsp.data,[4,3,1,2])
assert_array_equal(bsp.colind,[1,0,2,1])
assert_array_equal(bsp.indptr,[0,1,3,4])
assert_equal(bsp.getnnz(),4)
assert_equal(bsp.getformat(),'csr')
assert_array_almost_equal(bsp.todense(),b)
def check_constructor2(self):
b = zeros((6,6),'d')
b[3,4] = 5
bsp = csr_matrix(b)
assert_array_almost_equal(bsp.data,[5])
assert_array_equal(bsp.colind,[4])
assert_array_equal(bsp.indptr,[0,0,0,0,1,1,1])
assert_array_almost_equal(bsp.todense(),b)
def check_constructor3(self):
b = array([[1,0],
[0,2],
[3,0]],'d')
bsp = csr_matrix(b)
assert_array_almost_equal(bsp.data,[1,2,3])
assert_array_equal(bsp.colind,[0,1,0])
assert_array_equal(bsp.indptr,[0,1,2,3])
assert_array_almost_equal(bsp.todense(),b)
class test_csc(_test_cs):
spmatrix = csc_matrix
def check_constructor1(self):
b = array([[1,0,0],[3,0,1],[0,2,0]],'d')
bsp = csc_matrix(b)
assert_array_almost_equal(bsp.data,[1,3,2,1])
assert_array_equal(bsp.rowind,[0,1,2,1])
assert_array_equal(bsp.indptr,[0,2,3,4])
assert_equal(bsp.getnnz(),4)
assert_equal(bsp.getformat(),'csc')
def check_constructor2(self):
b = zeros((6,6),'d')
b[2,4] = 5
bsp = csc_matrix(b)
assert_array_almost_equal(bsp.data,[5])
assert_array_equal(bsp.rowind,[2])
assert_array_equal(bsp.indptr,[0,0,0,0,0,1,1])
def check_constructor3(self):
b = array([[1,0],[0,2],[3,0]],'d')
bsp = csc_matrix(b)
assert_array_almost_equal(bsp.data,[1,3,2])
assert_array_equal(bsp.rowind,[0,2,1])
assert_array_equal(bsp.indptr,[0,2,3])
if __name__ == "__main__":
ScipyTest('sparse.Sparse').run()
|
