fixup

JuliaLang · Oct 27, 2022 · ba31505 · ba31505
1 parent bf72c3d
commit ba31505
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/base/iterators.jl b/base/iterators.jl
@@ -946,7 +946,7 @@ true
 ```
 """
 cycle(xs) = Cycle(xs)
-cycle(xs, n::Int) = flatten(repeated(xs, n))
+cycle(xs, n::Integer) = flatten(repeated(xs, n))
 
 eltype(::Type{Cycle{I}}) where {I} = eltype(I)
 IteratorEltype(::Type{Cycle{I}}) where {I} = IteratorEltype(I)

diff --git a/test/iterators.jl b/test/iterators.jl
@@ -251,8 +251,13 @@ let i = 0
 end
 
 @testset "cycle(iter, n)" begin
- @test cycle(0:3, 2) == [0, 1, 2, 3, 0, 1, 2, 3]
+ @test collect(cycle(0:3, 2)) == [0, 1, 2, 3, 0, 1, 2, 3]
+ @test collect(cycle(Iterators.filter(iseven, 1:4), 2)) == [2, 4, 2, 4]
+ @test collect(take(cycle(countfrom(11), 3), 4)) == 11:14
+
  @test isempty(cycle(1:0)) == isempty(cycle(1:0, 3)) == true
+ @test isempty(cycle(1:5, 0))
+ @test isempty(cycle(Iterators.filter(iseven, 1:4), 0))
 
  @test eltype(cycle(0:3, 2)) === Int
  @test Base.IteratorEltype(cycle(0:3, 2)) == Base.HasEltype()