handle more linalg wrappers

#69

Author: mcabbott

src/functions.jl

@@ -294,6 +294,7 @@ Olist = [ :AbstractMatrix, :AbstractVector, :Number,
     :(Transpose{<:Any,<:AbstractMatrix}), :(Transpose{<:Any,<:AbstractVector}),
     :(NamedDimsArray{L,T,1} where {L,T}), :(NamedDimsArray{L,T,2} where {L,T}),
     :(Adjoint{<:Number,<:AbstractVector}), # 1.5 problem...
+    :(Diagonal), :(Union{LowerTriangular, UpperTriangular}),
     ]
 for (Ts, Ss) in [(Rlist, Rlist), (Rlist, Olist), (Olist, Rlist)]
     for T in Ts, S in Ss # some combinations are errors, later, that's ok

test/_functions.jl

@@ -1,4 +1,5 @@
 using Test, AxisKeys, Statistics
+using NamedDims: unname
 
 M = wrapdims(rand(Int8, 3,4), r='a':'c', c=2:5)
 MN = NamedDimsArray(M.data.data, r='a':'c', c=2:5)
@@ -267,6 +268,21 @@ end
 @testset "linalg" begin
     using LinearAlgebra
 
+    # more wrappers -- https://github.com/mcabbott/AxisKeys.jl/issues/69
+    @testset "wrappers" begin
+        D3 = Diagonal(randn(3))
+        U3 = UpperTriangular(rand(3,3))
+        U4 = UpperTriangular(rand(4,4) .+ im)
+
+        @test axiskeys(D3 * unname(M), 2) == 2:5
+        @test_skip U3 * M # *(::UpperTriangular, ::NamedDimsArray) is ambiguous
+        @test axiskeys(U4 * unname(M'), 2) == 'a':'c'
+
+        @test axiskeys(D3 \ M, 2) == 2:5
+        @test axiskeys(U3 \ M, 2) == 2:5
+        @test_skip U4 / M  # no method matching ldiv!(::NamedDims.NamedFactorization, ::Matrix{ComplexF64})
+    end
+
     @testset "cholesky" begin
         A = rand(10, 3)
         ka = KeyedArray(A, (0:9, ["a", "b", "c"]))