Fix corner case for diaglength

ITensor · Nov 30, 2023 · d5459a8 · d5459a8
1 parent ba43de9
commit d5459a8
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/NDTensors/src/lib/DiagonalArrays/src/diaginterface.jl b/NDTensors/src/lib/DiagonalArrays/src/diaginterface.jl
@@ -1,5 +1,7 @@
 using Compat: allequal
 
+diaglength(a::AbstractArray{<:Any,0}) = 1
+
 function diaglength(a::AbstractArray)
   return minimum(size(a))
 end

diff --git a/NDTensors/src/lib/DiagonalArrays/test/runtests.jl b/NDTensors/src/lib/DiagonalArrays/test/runtests.jl
@@ -9,5 +9,12 @@ using NDTensors.DiagonalArrays: DiagonalArrays
       ),
     ) isa Any
   end
+  @testset "Basics" begin
+    using NDTensors.DiagonalArrays: diaglength
+    a = fill(1.0, 2, 3)
+    @test diaglength(a) == 2
+    a = fill(1.0)
+    @test diaglength(a) == 1
+  end
 end
 end