Add mappedarrayreduce

Author: jishnub

src/MappedArrays.jl

@@ -2,7 +2,7 @@ module MappedArrays
 
 using Base: @propagate_inbounds
 
-export AbstractMappedArray, MappedArray, ReadonlyMappedArray, mappedarray, of_eltype
+export AbstractMappedArray, MappedArray, ReadonlyMappedArray, mappedarray, of_eltype, mappedarrayreduce
 
 abstract type AbstractMappedArray{T,N} <: AbstractArray{T,N} end
 abstract type AbstractMultiMappedArray{T,N} <: AbstractMappedArray{T,N} end
@@ -261,4 +261,9 @@ eltypes(A::AbstractArray) = Tuple{eltype(A)}
 ## Deprecations
 @deprecate mappedarray(f_finv::Tuple{Any,Any}, args::AbstractArray...) mappedarray(f_finv[1], f_finv[2], args...)
 
+
+# mapreduce
+
+mappedarrayreduce(f, op, A::AbstractArray...; kw...) = reduce(op, mappedarray(f, A...); kw...)
+
 end # module

test/runtests.jl

@@ -177,3 +177,23 @@ end
     str = String(take!(io))
     @test occursin("x1 + x2", str)
 end
+
+@testset "mapreduce" begin
+    for T in [Int, Float64]
+        x = rand(T, 10); y = similar(x);
+
+        f = x->x^2; op = +
+        @test mapreduce(f, op, x) == mappedarrayreduce(f, op, x)
+        @test mapreduce(f, op, x, init = zero(T)) == mappedarrayreduce(f, op, x, init = zero(T))
+        @test mapreduce(f, op, x, init = zero(T), dims = 1) == mappedarrayreduce(f, op, x, init = zero(T), dims = 1)
+        @test mapreduce(f, op, x, init = zero(T), dims = :) == mappedarrayreduce(f, op, x, init = zero(T), dims = :)
+
+        if VERSION >= v"1.2"
+            f = ==; op = +
+            @test mapreduce(f, op, x, y) == mappedarrayreduce(f, op, x, y)
+            @test mapreduce(f, op, x, y, init = 0) == mappedarrayreduce(f, op, x, y, init = 0)
+            @test mapreduce(f, op, x, y, init = 0, dims = 1) == mappedarrayreduce(f, op, x, y, init = 0, dims = 1)
+            @test mapreduce(f, op, x, y, init = 0, dims = :) == mappedarrayreduce(f, op, x, y, init = 0, dims = :)
+        end
+    end
+end