Skip to content

Commit b3009e5

Browse files
committed
test/simplicial: test SimplicialLLT::solve for sparse matrix
1 parent f728dfb commit b3009e5

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

unittest/python/decompositions/sparse/test_SimplicialLLT.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import numpy as np
2+
import scipy
23
from scipy.sparse import csc_matrix
34

45
import eigenpy
@@ -30,3 +31,14 @@
3031
llt.analyzePattern(A)
3132
llt.factorize(A)
3233
permutation = llt.permutationP()
34+
35+
X_sparse = scipy.sparse.random(dim, 10, random_state=rng)
36+
B_sparse = A.dot(X_sparse)
37+
B_sparse = B_sparse.tocsc(True)
38+
39+
if B_sparse.has_sorted_indices:
40+
B_sparse.sort_indices()
41+
42+
X_est = llt.solve(B_sparse)
43+
assert eigenpy.is_approx(X_est.toarray(), X_sparse.toarray())
44+
assert eigenpy.is_approx(A.dot(X_est.toarray()), B_sparse.toarray())

0 commit comments

Comments
 (0)