fix List iteration

and add a test
MikeInnes · Aug 27, 2018 · 66b4a76 · 66b4a76
1 parent 4324734
commit 66b4a76
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/Lazy.jl b/src/Lazy.jl
@@ -136,8 +136,7 @@ setindex!(xs::LinkedList, v, i::Integer) = i <= 1 ? xs.first = v : (tail(xs)[i-1
 setindex!(xs::LazyList, v, i::Integer) = i <= 1 ? realise(xs)[1] = v : (tail(xs)[i-1] = v)
 
 # Iteration over a list holds on to the head
-Base.iterate(xs::List) = xs, xs
-function Base.iterate(::List, xs::List)
+function Base.iterate(L::List, xs::List=L)
   isempty(xs) && return nothing
   first(xs), tail(xs)
 end

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -24,6 +24,7 @@ using Test
     @test list(1,2,list(3,4))[3] == list(3, 4)
     @test list(list(1), list(2))[1] == list(1)
     @test reductions(+, 0, list(1, 2, 3)) == list(1, 3, 6)
+    @test [i for i in @lazy[1,2,3]] == [1,2,3]
 end
 
 @testset "Fibs" begin