workersactive, nworkersactive and evenlyscatterproduct accept an Iterators.ProductIterator, added coverage badge

Author: jishnub

Project.toml

@@ -1,7 +1,7 @@
 name = "ParallelUtilities"
 uuid = "fad6cfc8-4f83-11e9-06cc-151124046ad0"
 authors = ["Jishnu Bhattacharya <[email protected]>"]
-version = "0.1.0"
+version = "0.1.1"
 
 [deps]
 Distributed = "8ba89e20-285c-5b6f-9357-94700520ee1b"

README.md

@@ -1,6 +1,7 @@
 # ParallelUtilities.jl
 
 [![Build Status](https://travis-ci.com/jishnub/ParallelUtilities.jl.svg?branch=master)](https://travis-ci.com/jishnub/ParallelUtilities.jl)
+[![Coverage Status](https://coveralls.io/repos/github/jishnub/ParallelUtilities.jl/badge.svg?branch=master)](https://coveralls.io/github/jishnub/ParallelUtilities.jl?branch=master)
 
 Parallel mapreduce and other helpful functions for HPC, meant primarily for embarassingly parallel operations that often require one to split up a list of tasks into subsections that can be processed on individual cores.
 

src/ParallelUtilities.jl

@@ -400,6 +400,9 @@ function evenlyscatterproduct(iterators::Tuple,
 	ProductSplit(iterators,np,procid)
 end
 
+evenlyscatterproduct(itp::Iterators.ProductIterator,args...) = 
+	evenlyscatterproduct(itp.iterators,args...)
+
 function whichproc(iterators::Tuple,val::Tuple,np::Int)
 
 	_infullrange(val,iterators) || return nothing
@@ -504,9 +507,11 @@ end
 	nt <= nw ? nt : nw
 end
 @inline nworkersactive(ps::ProductSplit) = nworkersactive(ps.iterators)
+@inline nworkersactive(itp::Iterators.ProductIterator) = nworkersactive(itp.iterators)
 @inline nworkersactive(args...) = nworkersactive(args)
 @inline workersactive(iterators::Tuple) = workers()[1:nworkersactive(iterators)]
 @inline workersactive(ps::ProductSplit) = workersactive(ps.iterators)
+@inline workersactive(itp::Iterators.ProductIterator) = workersactive(itp.iterators)
 @inline workersactive(args...) = workersactive(args)
 
 function gethostnames(procs_used = workers())

test/runtests.jl

@@ -181,6 +181,8 @@ end
         @test evenlyscatterproduct(n,np,proc_id) == ProductSplit((1:n,),np,proc_id)
         for iters in [(1:10,),(1:10,4:6),(1:10,4:6,1:4),(1:2:10,4:1:6)]
         	@test evenlyscatterproduct(iters,np,proc_id) == ProductSplit(iters,np,proc_id)
+        	itp = Iterators.product(iters...)
+        	@test evenlyscatterproduct(itp,np,proc_id) == ProductSplit(iters,np,proc_id)
         end
     end
 
@@ -281,6 +283,11 @@ end
 
         ps = ProductSplit((1:10,),nworkers(),1)
         @test nworkersactive(ps) == min(10,nworkers())
+
+        iters = (1:1,1:2)
+        itp = Iterators.product(iters...)
+        @test nworkersactive(itp) == nworkersactive(iters)
+        @test workersactive(itp) == workersactive(iters)
     end
 
     @testset "hostnames" begin