iteratorsize for partition iterator

Author: mschauer

base/iterator.jl

@@ -490,6 +490,9 @@ type PartitionIterator{T}
 end
 
 eltype{T}(::Type{PartitionIterator{T}}) = Vector{eltype(T)}
+partition_iteratorsize(::HasShape) = HasLength()
+partition_iteratorsize{T}(isz::T) = isz
+iteratorsize{T}(P::Type{PartitionIterator{T}}) = partition_iteratorsize(iteratorsize(T))
 
 function length(itr::PartitionIterator)
     l = length(itr.c)

test/functional.jl

@@ -264,10 +264,10 @@ let v = collect(Base.partition([1,2,3,4,5], 1))
     @test all(i->v[i][1] == i, v)
 end
 
-let v = collect(Base.partition([1,2,3,4,5], 2))
-    @test v[1] == [1,2]
-    @test v[2] == [3,4]
-    @test v[3] == [5]
+let v1 = collect(Base.partition([1,2,3,4,5], 2)), v2 = collect(Base.partition(Base.flatten([1,2,3,4,5]), 2))
+    @test v1[1] == v1[1] == [1,2]
+    @test v1[2] == v2[2] == [3,4]
+    @test v1[3] == v2[3] == [5]
 end
 
 let v = collect(Base.partition(enumerate([1,2,3,4,5]), 3))