Allow StepRange

Author: DanielVandH

Project.toml

@@ -1,7 +1,7 @@
 name = "HarmonicOrthogonalPolynomials"
 uuid = "e416a80e-9640-42f3-8df8-80a93ca01ea5"
 authors = ["Sheehan Olver <[email protected]>"]
-version = "0.6"
+version = "0.6.1"
 
 [deps]
 BlockArrays = "8e7c35d0-a365-5155-bbbb-fb81a777f24e"

src/multivariateops.jl

@@ -35,7 +35,7 @@ getindex(P::MultivariateOrthogonalPolynomial{D}, xy::StaticVector{D}, J::Block{1
 getindex(P::MultivariateOrthogonalPolynomial{D}, xy::StaticVector{D}, JR::BlockRange{1}) where D = P[xy, Block.(OneTo(Int(maximum(JR))))][JR]
 getindex(P::MultivariateOrthogonalPolynomial{D}, xy::StaticVector{D}, Jj::BlockIndex{1}) where D = P[xy, block(Jj)][blockindex(Jj)]
 getindex(P::MultivariateOrthogonalPolynomial{D}, xy::StaticVector{D}, j::Integer) where D = P[xy, findblockindex(axes(P,2), j)]
-getindex(P::MultivariateOrthogonalPolynomial{D}, xy::StaticVector{D}, jr::AbstractVector{<:Integer}) where D = P[xy, Block.(OneTo(Int(findblock(axes(P,2), maximum(jr)))))][jr]
+getindex(P::MultivariateOrthogonalPolynomial{D}, xy::StaticVector{D}, jr::AbstractVector{<:Integer}) where D = P[xy, Block.(OneTo(Integer(findblock(axes(P,2), maximum(jr)))))][jr]
 
 const FirstInclusion = BroadcastQuasiVector{<:Any, typeof(first), <:Tuple{Inclusion}}
 const LastInclusion = BroadcastQuasiVector{<:Any, typeof(last), <:Tuple{Inclusion}}

test/runtests.jl

@@ -428,3 +428,10 @@ end
     @test isdiag(A[1:N, 1:N])
     @test A[1:N, 1:N]^2 ≈ A2[1:N, 1:N]
 end
+
+@testset "Issue #88" begin
+    S = SphericalHarmonic()
+    v = S[SphericalCoordinate(0.1,0.2), 1:2:end]
+    @test length(v) == ∞
+    @test v[1:5] == S[SphericalCoordinate(0.1,0.2), 1:2:10]
+end