Overload map (#83)

* Overload map

* v0.8.4

Author: dlfivefifty

Project.toml

@@ -1,6 +1,6 @@
 name = "FillArrays"
 uuid = "1a297f60-69ca-5386-bcde-b61e274b549b"
-version = "0.8.3"
+version = "0.8.4"
 
 [deps]
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"

src/FillArrays.jl

@@ -4,7 +4,7 @@ using LinearAlgebra, SparseArrays
 import Base: size, getindex, setindex!, IndexStyle, checkbounds, convert,
     +, -, *, /, \, diff, sum, cumsum, maximum, minimum, sort, sort!,
     any, all, axes, isone, iterate, unique, allunique, permutedims, inv,
-    copy, vec, setindex!, count, ==, reshape, _throw_dmrs
+    copy, vec, setindex!, count, ==, reshape, _throw_dmrs, map
 
 import LinearAlgebra: rank, svdvals!, tril, triu, tril!, triu!, diag, transpose, adjoint, fill!, 
     norm2, norm1, normInf, normMinusInf, normp

src/fillbroadcast.jl

@@ -1,3 +1,7 @@
+### map
+
+map(f::Function, r::AbstractFill) = Fill(f(getindex_value(r)), axes(r))
+
 
 ### Unary broadcasting
 

test/runtests.jl

@@ -536,6 +536,18 @@ end
     end
 end
 
+@testset "map" begin
+    x = Ones(5)
+    @test map(exp,x) === Fill(exp(1.0),5)
+    @test map(isone,x) === Fill(true,5)
+
+    x = Zeros(5)
+    @test map(exp,x) === exp.(x)
+
+    x = Fill(2,5,3)
+    @test map(exp,x) === Fill(exp(2),5,3)
+end
+
 @testset "Sub-arrays" begin
     A = Fill(3.0,5)
     @test A[1:3] ≡ Fill(3.0,3)