Fix NamedArrayPartition similar when size is passed

Author: SouthEndMusic

src/named_array_partition.jl

@@ -33,8 +33,12 @@ end
 
 # return ArrayPartition when possible, otherwise next best thing of the correct size
 function Base.similar(A::NamedArrayPartition, dims::NTuple{N, Int}) where {N}
-    NamedArrayPartition(
-        similar(getfield(A, :array_partition), dims), getfield(A, :names_to_indices))
+    similar_array_partition = similar(getfield(A, :array_partition), dims)
+    if similar_array_partition isa ArrayPartition
+        NamedArrayPartition(similar_array_partition, getfield(A, :names_to_indices))
+    else
+        similar_array_partition
+    end
 end
 
 # similar array partition of common type

test/named_array_partition_tests.jl

@@ -6,6 +6,7 @@ using RecursiveArrayTools, Test
     @test typeof(x .^ 2) <: NamedArrayPartition
     @test typeof(similar(x)) <: NamedArrayPartition
     @test typeof(similar(x, Int)) <: NamedArrayPartition
+    @test typeof(similar(x, (5, 5))) <: Matrix
     @test x.a ≈ ones(10)
     @test typeof(x .+ x[1:end]) <: Vector # test broadcast precedence 
     @test all(x .== x[1:end])