Fix vecnorm(A::SparseMatrixCSC, ...).

JuliaLang · Oct 19, 2017 · 42c2963 · 42c2963
1 parent 01fc2dd
commit 42c2963
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/base/sparse/linalg.jl b/base/sparse/linalg.jl
@@ -514,7 +514,7 @@ end
 diff(a::SparseMatrixCSC, dim::Integer)= dim==1 ? sparse_diff1(a) : sparse_diff2(a)
 
 ## norm and rank
-vecnorm(A::SparseMatrixCSC, p::Real=2) = vecnorm(A.nzval, p)
+vecnorm(A::SparseMatrixCSC, p::Real=2) = vecnorm(view(A.nzval, 1:nnz(A)), p)
 
 function norm(A::SparseMatrixCSC,p::Real=2)
     m, n = size(A)

diff --git a/test/sparse/sparse.jl b/test/sparse/sparse.jl
@@ -1619,6 +1619,10 @@ end
     @test norm(Ai,1) ≈ norm(Array(Ai),1)
     @test norm(Ai,Inf) ≈ norm(Array(Ai),Inf)
     @test vecnorm(Ai) ≈ vecnorm(Array(Ai))
+    # make certain entries in nzval beyond
+    # the range specified in colptr do not
+    # impact vecnorm of a sparse matrix
+    @test (foo = speye(4); resize!(foo.nzval, 5)[5] = NaN; vecnorm(foo) == 2.0)
 end
 
 @testset "sparse matrix cond" begin