log for dense diagonal matrix with negative elements

Author: jishnub

src/dense.jl

@@ -914,8 +914,12 @@ julia> log(A)
 """
 function log(A::AbstractMatrix)
     # If possible, use diagonalization
-    if isdiag(A)
-        return applydiagonal(log, A)
+    if isdiag(A) && eltype(A) <: Union{Real,Complex}
+        if eltype(A) <: Real && all(>(0), diagview(A))
+            return applydiagonal(log, A)
+        else
+            return applydiagonal(log∘complex, A)
+        end
     elseif ishermitian(A)
         logHermA = log(Hermitian(A))
         PH = parent(logHermA)

test/dense.jl

@@ -1418,4 +1418,13 @@ end
     @test tril(GenericArray(M),-1) == res'
 end
 
+@testset "log for diagonal" begin
+    D = diagm([-2.0, 2.0])
+    @test log(D) ≈ log(UpperTriangular(D))
+    D = diagm([2.0, 2.0])
+    @test log(D) ≈ log(UpperTriangular(D))
+    D = diagm([2.0, 2.0*im])
+    @test log(D) ≈ log(UpperTriangular(D))
+end
+
 end # module TestDense