Merge branch 'master' into jishnub/triul

stdlib/LinearAlgebra/test/triangular.jl

@@ -925,4 +925,73 @@ end
  end
 end
 
+@testset "custom axes" begin
+ SZA = SizedArrays.SizedArray{(2,2)}([1 2; 3 4])
+ for T in (UpperTriangular, LowerTriangular, UnitUpperTriangular, UnitLowerTriangular)
+ S = T(SZA)
+ r = SizedArrays.SOneTo(2)
+ @test axes(S) === (r,r)
+ end
+end
+
+@testset "arithmetic with an immutable parent" begin
+ F = FillArrays.Fill(2, (4,4))
+ for UT in (UnitUpperTriangular, UnitLowerTriangular)
+ U = UT(F)
+ @test -U == -Array(U)
+ end
+
+ F = FillArrays.Fill(3im, (4,4))
+ for U in (UnitUpperTriangular(F), UnitLowerTriangular(F))
+ @test imag(F) == imag(collect(F))
+ end
+end
+
+@testset "error paths" begin
+ A = zeros(1,1); B = zeros(2,2)
+ @testset "inplace mul scaling with incompatible sizes" begin
+ for T in (UpperTriangular, LowerTriangular, UnitUpperTriangular, UnitLowerTriangular)
+ @test_throws DimensionMismatch mul!(T(A), T(B), 3)
+ @test_throws DimensionMismatch mul!(T(A), 3, T(B))
+ end
+ end
+ @testset "copyto with incompatible sizes" begin
+ for T in (UpperTriangular, LowerTriangular, UnitUpperTriangular, UnitLowerTriangular)
+ @test_throws BoundsError copyto!(T(A), T(B))
+ end
+ end
+end
+
+@testset "arithmetic with partly uninitialized matrices" begin
+ @testset "$(typeof(A))" for A in (Matrix{BigFloat}(undef,2,2), Matrix{Complex{BigFloat}}(undef,2,2)')
+ A[1,1] = A[2,2] = A[2,1] = 4
+ B = Matrix{eltype(A)}(undef, size(A))
+ for MT in (LowerTriangular, UnitLowerTriangular)
+ L = MT(A)
+ B .= 0
+ copyto!(B, L)
+ @test L * 2 == 2 * L == 2B
+ @test L/2 == B/2
+ @test 2\L == 2\B
+ @test real(L) == real(B)
+ @test imag(L) == imag(B)
+ end
+ end
+
+ @testset "$(typeof(A))" for A in (Matrix{BigFloat}(undef,2,2), Matrix{Complex{BigFloat}}(undef,2,2)')
+ A[1,1] = A[2,2] = A[1,2] = 4
+ B = Matrix{eltype(A)}(undef, size(A))
+ for MT in (UpperTriangular, UnitUpperTriangular)
+ U = MT(A)
+ B .= 0
+ copyto!(B, U)
+ @test U * 2 == 2 * U == 2B
+ @test U/2 == B/2
+ @test 2\U == 2\B
+ @test real(U) == real(B)
+ @test imag(U) == imag(B)
+ end
+ end
+end
+
 end # module TestTriangular