reduced_indices for Tuple{Vararg{SOneTo}} (#1027)

Author: jishnub

src/SOneTo.jl

@@ -74,3 +74,7 @@ end
 
 Base.promote_rule(a::Type{Base.OneTo{T}}, ::Type{SOneTo{n}}) where {T,n} =
     Base.OneTo{promote_type(T, Int)}
+
+function Base.reduced_indices(inds::Tuple{SOneTo,Vararg{SOneTo}}, d::Int)
+    Base.reduced_indices(map(Base.OneTo, inds), d)
+end

test/abstractarray.jl

@@ -234,6 +234,15 @@ using StaticArrays, Test, LinearAlgebra
             @test (@inferred f(v, i)) == length(v[i])
         end
     end
+
+    @testset "reduced_indices" begin
+        s = SArray{Tuple{2,2,2},Int,3,8}((1,2,3,4,5,6,7,8))
+        a = Array(s)
+        for i in 1:ndims(s)
+            rs = @inferred Base.reduced_indices(axes(s), i)
+            @test rs == Base.reduced_indices(axes(a), i)
+        end
+    end
 end
 
 @testset "permutedims" begin