Implement Base.resize! for ArrayPartitions

Author: efaulhaber

src/array_partition.jl

@@ -124,6 +124,12 @@ Base.ones(A::ArrayPartition, dims::NTuple{N, Int}) where {N} = ones(A)
     return :($res)
 end
 
+## resize!
+function Base.resize!(A::ArrayPartition, sizes::Tuple)
+    resize!.(A.x, Ref(sizes))
+    A
+end
+
 ## vector space operations
 
 for op in (:+, :-)

test/partitions_test.jl

@@ -60,6 +60,10 @@ copyto!(p, c)
 @test c[1:5] == p.x[1]
 @test c[6:10] == p.x[2]
 
+resize!(p, (6, 7))
+@test length(p.x[1]) == 6
+@test length(p.x[2]) == 7
+
 ## inference tests
 
 x = ArrayPartition([1, 2], [3.0, 4.0])