Add one and copy for Diagonal (#52)

putianyi889 · dlfivefifty · web-flow · commit 058032d4bc19 · 2021-03-07T11:45:36.000Z
* Update infarrays.jl

* Update runtests.jl

* Update infarrays.jl

* Update runtests.jl

* Update infarrays.jl

Co-authored-by: Sheehan Olver &lt;solver@mac.com&gt;
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
