| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051 |
- """test sparse matrix construction functions"""
- from numpy.testing import assert_equal
- from scipy.sparse import csr_matrix, csr_array, sparray
- import numpy as np
- from scipy.sparse import _extract
- class TestExtract:
- def setup_method(self):
- self.cases = [
- csr_array([[1,2]]),
- csr_array([[1,0]]),
- csr_array([[0,0]]),
- csr_array([[1],[2]]),
- csr_array([[1],[0]]),
- csr_array([[0],[0]]),
- csr_array([[1,2],[3,4]]),
- csr_array([[0,1],[0,0]]),
- csr_array([[0,0],[1,0]]),
- csr_array([[0,0],[0,0]]),
- csr_array([[1,2,0,0,3],[4,5,0,6,7],[0,0,8,9,0]]),
- csr_array([[1,2,0,0,3],[4,5,0,6,7],[0,0,8,9,0]]).T,
- ]
- def test_find(self):
- for A in self.cases:
- I,J,V = _extract.find(A)
- B = csr_array((V,(I,J)), shape=A.shape)
- assert_equal(A.toarray(), B.toarray())
- def test_tril(self):
- for A in self.cases:
- B = A.toarray()
- for k in [-3,-2,-1,0,1,2,3]:
- assert_equal(_extract.tril(A,k=k).toarray(), np.tril(B,k=k))
- def test_triu(self):
- for A in self.cases:
- B = A.toarray()
- for k in [-3,-2,-1,0,1,2,3]:
- assert_equal(_extract.triu(A,k=k).toarray(), np.triu(B,k=k))
- def test_array_vs_matrix(self):
- for A in self.cases:
- assert isinstance(_extract.tril(A), sparray)
- assert isinstance(_extract.triu(A), sparray)
- M = csr_matrix(A)
- assert not isinstance(_extract.tril(M), sparray)
- assert not isinstance(_extract.triu(M), sparray)
|
