Merge branch 'mfalt-patch-1'

Author: jverzani

src/Polynomials.jl

@@ -193,6 +193,7 @@ norm(q::Poly, args...) = norm(coeffs(q), args...)
 
 """
 getindex{T}(p::Poly{T}, i) = (i+1 > length(p.a) ? zero(T) : p.a[i+1])
+getindex{T}(p::Poly{T}, idx::AbstractArray) = map(i->p[i], idx)
 function setindex!(p::Poly, v, i)
     n = length(p.a)
     if n < i+1
@@ -202,6 +203,10 @@ function setindex!(p::Poly, v, i)
     p.a[i+1] = v
     v
 end
+function setindex!(p::Poly, vs, idx::AbstractArray)
+    [setindex!(p, v, i) for (i,v) in zip(idx, vs)]
+    p
+end
 
 copy(p::Poly) = Poly(copy(p.a), p.var)
 

test/runtests.jl

@@ -163,7 +163,7 @@ p1[5] = 1
 @test p1 == Poly([1,2,1,0,0,1])
 
 
-## elementwise operations
+## elementwise operations #52
 println("Test for element-wise operations")
 p1  = Poly([1, 2])
 p2  = Poly([3, 1.])
@@ -177,3 +177,9 @@ pmin  = p-3
 @test isa(pprod,Vector{Poly{Float64}})
 @test isa(pmin, Vector{Poly{Float64}})
 
+## getindex with ranges #43
+p1 = Poly([4,5,6])
+@test all(p1[0:1] .== [4,5])
+@test all(p1[0:end] .== [4,5,6])
+p1[0:1] = [7,8]
+@test all(p1[0:end] .== [7,8,6])