Implement iterate properly

Sagnac · Jun 1, 2023 · 83274dd · 83274dd
1 parent 01e269b
commit 83274dd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/WavefrontError.jl b/src/WavefrontError.jl
@@ -142,7 +142,7 @@ end
 getindex(W::T, i) where {T <: WavefrontOutput} = getfield(W, fieldnames(T)[i])
 
 # hook into iterate to allow non-property destructuring of the output
-iterate(W::WavefrontOutput, i = 1) = (W[i], i + 1)
+iterate(W::WavefrontOutput, i = 1) = (i > 4 ? nothing : (W[i], i + 1))
 
 # methods
 function W(ρ::Vector, θ::Vector, OPD::Vector, n_max::Int, ::Model; precision = 3)

diff --git a/src/Zernike.jl b/src/Zernike.jl
@@ -208,7 +208,7 @@ show(::IO, ::Output) = nothing
 getindex(Z::T, i) where {T <: Output} = getfield(Z, fieldnames(T)[i])
 
 # hook into iterate to allow non-property destructuring of the output
-iterate(Z::Output, i = 1) = (Z[i], i + 1)
+iterate(Z::Output, i = 1) = (i > 3 ? nothing : (Z[i], i + 1))
 
 # methods
 function Z(m::Int, n::Int, ::Model)