return nothing in whichproc_localindex (#12)

jishnub · Jun 6, 2021 · 9ea2827 · 9ea2827
1 parent 2f26946
commit 9ea2827
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/src/productsplit.jl b/src/productsplit.jl
@@ -906,7 +906,9 @@ julia> ParallelUtilities.ProductSplit(iters, np, 4) |> collect
 """
 function whichproc_localindex(iterators::Tuple{Vararg{AbstractRange}}, val::Tuple, np::Integer)
     procid = whichproc(iterators, val, np)
+    procid === nothing && return nothing
     index = localindex(ProductSplit(iterators, np, procid), val)
+    index === nothing && return nothing
     return procid, index
 end
 

diff --git a/test/productsplit.jl b/test/productsplit.jl
@@ -337,6 +337,7 @@ end
                     end
                 end
             end
+            @test whichproc_localindex((1:1,1:1), (1,2), 1) === nothing
         end
 
         @testset "getindex" begin