Generalise last for broadcastVector (#359)

* Generalise last for broadcastVector

* Update broadcasttests.jl

* v2.3.2

Author: dlfivefifty

Project.toml

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

src/lazybroadcasting.jl

@@ -88,7 +88,10 @@ Broadcasted(A::SubArray{<:Any,N,<:BroadcastArray}) where N = broadcasted(A)::Bro
 axes(A::BroadcastArray) = axes(broadcasted(A))
 size(A::BroadcastArray) = map(length, axes(A))
 
-last(A::BroadcastArray) = A.f(last.(A.args)...)
+_broadcast_last(a) = a
+_broadcast_last(a::AbstractArray) = last(a)
+_broadcast_last(a::Ref) = a[]
+last(A::BroadcastArray) = A.f(_broadcast_last.(A.args)...)
 
 @propagate_inbounds getindex(A::BroadcastArray{T,N}, kj::Vararg{Int,N}) where {T,N} = convert(T,broadcasted(A)[kj...])::T
 

test/broadcasttests.jl

@@ -1,6 +1,7 @@
 module BroadcastTests
 
-using LazyArrays, ArrayLayouts, LinearAlgebra, FillArrays, StaticArrays, Tracker, Base64, Test
+using LazyArrays, ArrayLayouts, LinearAlgebra, FillArrays, Base64, Test
+using StaticArrays, Tracker
 import LazyArrays: BroadcastLayout, arguments, LazyArrayStyle, sub_materialize
 import Base: broadcasted
 
@@ -415,6 +416,11 @@ using Infinities
         @test transpose(b)[1:5] == exp.((1:5) .+ im)
         @test b'[1:5] == exp.((1:5) .- im)
     end
+
+    @testset "BroadcastVector last" begin
+        a = BroadcastArray(Base.literal_pow, ^, 1:5, Val(2))
+        @test last(a) == 25
+    end
 end
 
 end #module