use channel to make test more robust

Author: penelopeysm

test/sample.jl

@@ -681,14 +681,16 @@
 
     @testset "chain_number keyword argument" begin
         @testset for m in [MCMCSerial(), MCMCThreads(), MCMCDistributed()]
-            # check that the `chain_number` keyword argument is passed to the callback
-            chain_numbers = Int[]
-            function callback(args...; kwargs...)
-                @test haskey(kwargs, :chain_number)
-                return push!(chain_numbers, kwargs[:chain_number])
-            end
             niters = 10
-            chain = sample(MyModel(), MySampler(), m, niters, 4; callback=callback)
+            channel = Channel{Int}() do chn
+                # check that the `chain_number` keyword argument is passed to the callback
+                function callback(args...; kwargs...)
+                    @test haskey(kwargs, :chain_number)
+                    return put!(chn, kwargs[:chain_number])
+                end
+                chain = sample(MyModel(), MySampler(), m, niters, 4; callback=callback)
+            end
+            chain_numbers = collect(channel)
             # if the test fails, the output is too long to show, so print it manually to see
             # what's going on
             if sort(chain_numbers) != repeat(1:4; inner=niters)