Merge pull request #363 from AayushSabharwal/as/arraypartition-length

fix: fix `Base.length` for empty `ArrayPartition`, add tests

Author: ChrisRackauckas

src/array_partition.jl

@@ -285,7 +285,7 @@ recursive_eltype(A::ArrayPartition) = recursive_eltype(first(A.x))
 Base.iterate(A::ArrayPartition) = iterate(Chain(A.x))
 Base.iterate(A::ArrayPartition, state) = iterate(Chain(A.x), state)
 
-Base.length(A::ArrayPartition) = sum(broadcast(length, A.x))
+Base.length(A::ArrayPartition) = sum(broadcast(length, A.x); init = 0)
 Base.size(A::ArrayPartition) = (length(A),)
 
 # redefine first and last to avoid slow and not type-stable indexing

test/partitions_test.jl

@@ -1,4 +1,8 @@
 using RecursiveArrayTools, Test, Statistics, ArrayInterface
+
+@test length(ArrayPartition()) == 0
+@test isempty(ArrayPartition())
+
 A = (rand(5), rand(5))
 p = ArrayPartition(A)
 @inferred p[1]