Make static_length support StaticArrays

Author: chriselrod

Project.toml

@@ -1,6 +1,6 @@
 name = "ArrayInterface"
 uuid = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"
-version = "2.14.11"
+version = "2.14.12"
 
 [deps]
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"

src/ArrayInterface.jl

@@ -790,6 +790,8 @@ function __init__()
         known_first(::Type{<:StaticArrays.SOneTo}) = 1
         known_last(::Type{StaticArrays.SOneTo{N}}) where {N} = N
         known_length(::Type{StaticArrays.SOneTo{N}}) where {N} = N
+        known_length(::Type{StaticArrays.Length{L}}) where {L} = L
+        known_length(::Type{A}) where {A <: StaticArrays.StaticArray} = known_length(StaticArrays.Length(A))
 
         device(::Type{<:StaticArrays.MArray}) = CPUPointer()
         contiguous_axis(::Type{<:StaticArrays.StaticArray}) = Contiguous{1}()

test/runtests.jl

@@ -508,13 +508,20 @@ end
     @test ArrayInterface.known_length((1,)) == 1
     @test ArrayInterface.known_length((a=1,b=2)) == 2
     @test ArrayInterface.known_length([]) == nothing
-
+    
     x = view(SArray{Tuple{3,3,3}}(ones(3,3,3)), :, SOneTo(2), 2)
     @test @inferred(ArrayInterface.known_length(x)) == 6
     @test @inferred(ArrayInterface.known_length(x')) == 6
 
     itr = StaticInt(1):StaticInt(10)
     @inferred(ArrayInterface.known_length((i for i in itr))) == 10
+
+    v = @SVector rand(8);
+    A = @MMatrix rand(7,6);
+    T = SizedArray{Tuple{5,4,3}}(zeros(5,4,3));
+    @test @inferred(ArrayInterface.static_length(v)) === StaticInt(8)
+    @test @inferred(ArrayInterface.static_length(A)) === StaticInt(42)
+    @test @inferred(ArrayInterface.static_length(T)) === StaticInt(60)
 end
 
 @testset "indices" begin