permutedims for BroadcastArray (#227)

Author: dlfivefifty

Project.toml

@@ -1,6 +1,6 @@
 name = "LazyArrays"
 uuid = "5078a376-72f3-5289-bfd5-ec5146d43c02"
-version = "0.22.14"
+version = "0.22.15"
 
 [deps]
 ArrayLayouts = "4c555306-a7a7-4459-81d9-ec55ddd5c99a"

src/lazybroadcasting.jl

@@ -391,4 +391,6 @@ for op in (:*, :\, :/)
         getindex(A::BroadcastMatrix{<:Any,typeof($op),<:Tuple{AbstractVector,AbstractMatrix}}, ::Colon, j::Integer) = broadcast($op, A.args[1], A.args[2][:,j])
         getindex(A::BroadcastMatrix{<:Any,typeof($op),<:Tuple{AbstractMatrix,AbstractVector}}, ::Colon, j::Integer) = broadcast($op, A.args[1][:,j], A.args[2])
     end
-end
+end
+
+permutedims(A::BroadcastArray{T}) where T = BroadcastArray{T}(A.f, map(_permutedims,A.args)...)

test/broadcasttests.jl

@@ -339,4 +339,13 @@ import Base: broadcasted
             @test rowsupport(A, 3) == 3:5
         end
     end
+
+    @testset "permutedims" begin
+        A = BroadcastArray(exp, randn(5,5) .+ im)
+        @test permutedims(A) ≈ transpose(A)
+        B = BroadcastArray(+, randn(5,5) .+ im, 1)
+        @test permutedims(B) ≈ transpose(B)
+        C = BroadcastArray(+, randn(5,5) .+ im, randn(5))
+        @test permutedims(C) ≈ transpose(C)
+    end
 end