add where and tests.

Author: chriselrod

src/stridelayout.jl

@@ -268,12 +268,12 @@ end
 
 @inline size(B::Union{Transpose{T,A},Adjoint{T,A}}) where {T,A<:AbstractMatrix{T}} = permute(size(parent(B)), Val{(2,1)}())
 @inline size(B::PermutedDimsArray{T,N,I1,I2,A}) where {T,N,I1,I2,A<:AbstractArray{T,N}} = permute(size(parent(B)), Val{I1}())
-@inline size(A::AbstractArray, ::Static{N}) = size(A)[N]
-@inline size(A::AbstractArray, ::Val{N}) = size(A)[N]
+@inline size(A::AbstractArray, ::Static{N}) where {N} = size(A)[N]
+@inline size(A::AbstractArray, ::Val{N}) where {N} = size(A)[N]
 @inline strides(B::Union{Transpose{T,A},Adjoint{T,A}}) where {T,A<:AbstractMatrix{T}} = permute(strides(parent(B)), Val{(2,1)}())
 @inline strides(B::PermutedDimsArray{T,N,I1,I2,A}) where {T,N,I1,I2,A<:AbstractArray{T,N}} = permute(strides(parent(B)), Val{I1}())
-@inline stride(A::AbstractArray, ::Static{N}) = stride(A)[N]
-@inline stride(A::AbstractArray, ::Val{N}) = stride(A)[N]
+@inline stride(A::AbstractArray, ::Static{N}) where {N} = strides(A)[N]
+@inline stride(A::AbstractArray, ::Val{N}) where {N} = strides(A)[N]
 
 size(B::S) where {N,NP,T,A<:AbstractArray{T,NP},I,S <: SubArray{T,N,A,I}} = _size(size(parent(B)), B.indices, map(static_length, B.indices))
 strides(B::S) where {N,NP,T,A<:AbstractArray{T,NP},I,S <: SubArray{T,N,A,I}} = _strides(strides(parent(B)), B.indices)

test/runtests.jl

@@ -290,6 +290,9 @@ using OffsetArrays
     @test @inferred(ArrayInterface.size(S)) == size(S)
     @test @inferred(ArrayInterface.size(Sp)) == size(Sp)
     @test @inferred(ArrayInterface.size(Sp2)) == size(Sp2)
+    @test @inferred(ArrayInterface.size(Sp2, Static(1))) === 2
+    @test @inferred(ArrayInterface.size(Sp2, Static(2))) === Static(3)
+    @test @inferred(ArrayInterface.size(Sp2, Static(3))) === Static(2)
 
     @test @inferred(ArrayInterface.size(M)) === (Static(2), Static(3), Static(4))
     @test @inferred(ArrayInterface.size(Mp)) === (Static(3), Static(4))
@@ -306,6 +309,9 @@ using OffsetArrays
     @test @inferred(ArrayInterface.strides(S)) === (Static(1), Static(2), Static(6))
     @test @inferred(ArrayInterface.strides(Sp)) === (Static(6), Static(1), Static(2))
     @test @inferred(ArrayInterface.strides(Sp2)) === (Static(6), Static(2), Static(1))
+    @test @inferred(ArrayInterface.stride(Sp2, Static(1))) === Static(6)
+    @test @inferred(ArrayInterface.stride(Sp2, Static(2))) === Static(2)
+    @test @inferred(ArrayInterface.stride(Sp2, Static(3))) === Static(1)
 
     @test @inferred(ArrayInterface.strides(M)) === (Static(1), Static(2), Static(6))
     @test @inferred(ArrayInterface.strides(Mp)) === (Static(2), Static(6))