Fixes Edge Cases with #167 (#169)

* Fixes Edge Cases with #167

Author: JosephTomlinson

src/darray.jl

@@ -285,7 +285,18 @@ end
 # get array of start indices for dividing sz into nc chunks
 function defaultdist(sz::Int, nc::Int)
     if sz >= nc
-        return ceil.(Int, range(1, stop=sz+1, length=nc+1))
+        chunk_size = div(sz,nc)
+        remainder = rem(sz,nc)
+        grid = zeros(Int64, nc+1)
+        for i = 1:(nc+1)
+            grid[i] += (i-1)*chunk_size + 1
+            if i<= remainder
+                grid[i] += i-1
+            else
+                grid[i] += remainder
+            end
+        end
+        return grid
     else
         return [[1:(sz+1);]; zeros(Int, nc-sz)]
     end

test/darray.jl

@@ -63,7 +63,10 @@ using Random
         @test fetch(@spawnat MYID length(localpart(DA)) == 2)
         @test fetch(@spawnat OTHERIDS length(localpart(DA)) == 1)
         close(DA)
+        @test DistributedArrays.defaultdist(50,4) == [1,14,27,39,51]
     end
+    
+    
 end
 
 check_leaks()