Sum over an iterator (#195)

Author: hayatoikoma

src/array.jl

@@ -106,6 +106,13 @@ function linspace{T}(::Type{AFArray}, a::T, b::T, c::Integer)
     dx = (b_fl - a_fl)/(Float64(c) - 1.0)
     range(AFArray{Float64}, a_fl, dx, c)
 end
+function Base.sum(A::AFArray, ndims)
+    B = A
+    for i in ndims
+        B = sum(B, i)
+    end
+    return B
+end
 isfinite(a::AFArray) = !isinf(a) & !isnan(a)
 
 import Base: /, *, +, -, ^, ==, <, >, <=, >=, !, !=, &, |, <<, >>, xor

test/math.jl

@@ -144,6 +144,10 @@ amf = AFArray(am)
 
 @test Array(sum(amf, 1)) ≈ sum(am, 1)
 @test Array(sum(amf, 2)) ≈ sum(am, 2)
+@test Array(sum(amf, [1,2])) ≈ sum(am, [1,2])
+@test Array(sum(amf, [2,1])) ≈ sum(am, [2,1])
+@test Array(sum(amf, 1:2)) ≈ sum(am, 1:2)
+@test Array(sum(amf, (1,2))) ≈ sum(am, (1,2))
 
 @test Array(prod(amf, 1)) ≈ prod(am, 1)
 @test Array(prod(amf, 2)) ≈ prod(am, 2)