Skip to content

Commit 90dd4c7

Browse files
committed
test/sparse: test full zero matrices
1 parent f528d72 commit 90dd4c7

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

unittest/python/test_sparse_matrix.py

+6-2
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,7 @@
1616
v = sparse_matrix.vector1x1(2)
1717
assert v.toarray() == np.array([2])
1818

19-
size = 100
20-
diag_values = np.random.rand(100)
19+
diag_values = np.random.rand(10)
2120
diag_mat = sparse_matrix.diagonal(diag_values)
2221
assert (diag_mat.toarray() == np.diag(diag_values)).all()
2322

@@ -26,3 +25,8 @@
2625

2726
diag_mat_csr = csr_matrix(diag_mat)
2827
assert (sparse_matrix.copy(diag_mat_csr) != diag_mat_csr).nnz == 0
28+
29+
# test zero matrix
30+
zero_mat = csr_matrix(np.zeros((10, 1)))
31+
zero_mat_copy = sparse_matrix.copy(zero_mat)
32+
assert (zero_mat_copy != zero_mat).nnz == 0

0 commit comments

Comments
 (0)