Minor code refactor and more test cases for TArray (#49)

Author: KDr2

src/tarray.jl

@@ -53,7 +53,7 @@ function Base.getindex(S::TArray{T, N}, I::Vararg{Int,N}) where {T, N}
     return d[I...]
 end
 
-function Base.setindex!(S::TArray{T, N}, x, I::Vararg{Int,N}) where {T, N}
+function Base.setindex!(S::TArray{T, N}, x::T, I::Vararg{Int,N}) where {T, N}
     n, d = task_local_storage(S.ref)
     cn   = n_copies()
     newd = d
@@ -65,7 +65,7 @@ function Base.setindex!(S::TArray{T, N}, x, I::Vararg{Int,N}) where {T, N}
     newd[I...] = x
 end
 
-function Base.push!(S::TArray, x)
+function Base.push!(S::TArray{T}, x::T) where T
     n, d = task_local_storage(S.ref)
     cn   = n_copies()
     newd = d
@@ -102,21 +102,27 @@ end
 
 function Base.display(S::TArray)
     arr = S.orig_task.storage[S.ref][2]
-    @warn "display(::TArray) prints the originating task's storage, not the current task's storage. Please use show(::TArray) to display the current task's version of a TArray."
+    @warn "display(::TArray) prints the originating task's storage, " *
+        "not the current task's storage. " *
+        "Please use show(::TArray) to display the current task's version of a TArray."
     display(arr)
 end
 
 Base.show(io::IO, S::TArray) = Base.show(io::IO, task_local_storage(S.ref)[2])
-Base.size(S::TArray) = Base.size(task_local_storage(S.ref)[2])
-Base.ndims(S::TArray) = Base.ndims(task_local_storage(S.ref)[2])
 
 # Base.get(t::Task, S) = S
 # Base.get(t::Task, S::TArray) = (t.storage[S.ref][2])
 Base.get(S::TArray) = (current_task().storage[S.ref][2])
 
-# Implements eltype, firstindex, lastindex, and iterate
-# functions.
-for F in (:eltype, :firstindex, :lastindex, :iterate)
+##
+# Iterator Interface
+IteratorSize(::Type{TArray{T, N}}) where {T, N} = HasShape{N}()
+IteratorEltype(::Type{TArray}) = HasEltype()
+
+# Implements iterate, eltype, length, and size functions,
+# as well as firstindex, lastindex, ndims, and axes
+for F in (:iterate, :eltype, :length, :size,
+          :firstindex, :lastindex, :ndims, :axes)
     @eval Base.$F(a::TArray, args...) = $F(get(a), args...)
 end
 

src/tref.jl

@@ -1,5 +1,5 @@
 ##########
-# TRef #
+#  TRef  #
 ##########
 
 """

test/runtests.jl

@@ -2,3 +2,4 @@ include("clonetask.jl")
 include("brokentask.jl")
 include("tarray.jl")
 include("tarray2.jl")
+include("tarray3.jl")

test/tarray3.jl

@@ -0,0 +1,32 @@
+using Libtask
+using Test
+
+@testset "TArray Copying Test" begin
+    DATA = Dict{Task, Array}()
+
+    function f_ct()
+        ta = tzeros(UInt64, 4);
+        for i in 1:4
+            ta[i] = hash(current_task())
+            DATA[current_task()] = convert(Array, ta)
+            produce(ta[i])
+        end
+    end
+
+    t = CTask(f_ct)
+
+    @test consume(t) == hash(t) # index = 1
+    @test consume(t) == hash(t) # index = 2
+
+    a = copy(t);
+
+    @test consume(a) == hash(a) # index = 3
+    @test consume(a) == hash(a) # index = 4
+
+    @test consume(t) == hash(t) # index = 3
+
+
+    @test DATA[t] == [hash(t), hash(t), hash(t), 0]
+    @test DATA[a] == [hash(t), hash(t), hash(a), hash(a)]
+
+end