Add one and copy for Diagonal (#52)

* Update infarrays.jl * Update runtests.jl * Update infarrays.jl * Update runtests.jl * Update infarrays.jl Co-authored-by: Sheehan Olver <[email protected]>
JuliaArrays · Mar 7, 2021 · 058032d · 058032d · dlfivefifty · Mar 7, 2021
1 parent a9437da
commit 058032d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/infarrays.jl b/src/infarrays.jl
@@ -234,6 +234,8 @@ end
 # Diagonal
 #####
 
+one(D::Diagonal{T,<:AbstractFill{T,1,Tuple{OneToInf{Int}}}}) where T = Eye{T}(size(D,1))
+copy(D::Diagonal{T,<:AbstractFill{T,1,Tuple{OneToInf{Int}}}}) where T = D
 
 BroadcastStyle(::Type{<:Diagonal{<:Any,<:AbstractInfUnitRange}}) = LazyArrayStyle{2}()
 

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -517,8 +517,12 @@ end
 
     @test Eye{Int}(∞) * D ≡ Eye{Int}(∞) * D ≡ D
     @test Eye(∞) * D == Eye(∞) * D == D
+    @test Eye(∞) == Eye(∞)^0 == Eye(∞)^1 == Eye(∞)^2 == one(Eye(∞)) == copy(Eye(∞)) == one(Diagonal(Fill(2,∞)))
+    @test Diagonal(Fill(2,∞)) == copy(Diagonal(Fill(2,∞)))
+
 
     @test permutedims(D) ≡ D
+
 end
 
 @testset "concat" begin